Commit 6298ab02 authored by MasterScrat's avatar MasterScrat
Browse files

Checking timing skipping all episodes

parent d26e2719
......@@ -25,11 +25,15 @@ from utils.observation_utils import normalize_observation
# Checkpoint to use
checkpoint = "checkpoints/sample-checkpoint.pth"
# Beyond this number of agents, skip the episode
max_num_agents_handled = 0
# Observation parameters
# These need to match your training parameters!
observation_tree_depth = 1
observation_radius = 10
observation_max_path_depth = 20
##########################
remote_client = FlatlandRemoteClient()
......@@ -108,7 +112,7 @@ while True:
steps += 1
obs_time, agent_time, step_time = 0.0, 0.0, 0.0
if nb_agents < 100 and not check_if_all_blocked(env=local_env):
if nb_agents <= max_num_agents_handled:# and not check_if_all_blocked(env=local_env):
time_start = time.time()
action = rl_controller(observation, nb_agents)
agent_time = time.time() - time_start
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment