Skip to content
Snippets Groups Projects
Commit 85a29894 authored by Erik Nygren's avatar Erik Nygren :bullettrain_front:
Browse files

Changed renderEnv to render_env

parent dc738d78
No related branches found
No related tags found
No related merge requests found
# Default ignored files
/workspace.xml
\ No newline at end of file
......@@ -3,13 +3,13 @@ from collections import deque
import numpy as np
import torch
from flatland.envs.generators import rail_from_file, complex_rail_generator
from observation_builders.observations import TreeObsForRailEnv
from predictors.predictions import ShortestPathPredictorForRailEnv
from flatland.envs.generators import rail_from_file
from flatland.envs.rail_env import RailEnv
from flatland.utils.rendertools import RenderTool
from importlib_resources import path
import time
from observation_builders.observations import TreeObsForRailEnv
from predictors.predictions import ShortestPathPredictorForRailEnv
import torch_training.Nets
from torch_training.dueling_double_dqn import Agent
from utils.observation_utils import normalize_observation
......@@ -73,7 +73,7 @@ action_prob = [0] * action_size
agent_obs = [None] * env.get_num_agents()
agent_next_obs = [None] * env.get_num_agents()
agent = Agent(state_size, action_size, "FC", 0)
with path(torch_training.Nets, "avoid_checkpoint59900.pth") as file_in:
with path(torch_training.Nets, "avoid_checkpoint60000.pth") as file_in:
agent.qnetwork_local.load_state_dict(torch.load(file_in))
record_images = False
......
......@@ -118,7 +118,7 @@ def main(argv):
# Only render when not triaing
if not Training:
env_renderer.renderEnv(show=True, show_observations=True)
env_renderer.render_env(show=True, show_observations=True)
# Chose the actions
for a in range(env.get_num_agents()):
......@@ -210,7 +210,7 @@ def main(argv):
# Run episode
for step in range(max_steps):
env_renderer.renderEnv(show=True, show_observations=False)
env_renderer.render_env(show=True, show_observations=False)
# Chose the actions
for a in range(env.get_num_agents()):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment