Commit 73633075 authored by MasterScrat's avatar MasterScrat
Browse files

Non-debug submission on more environments

parent d63ccbdd
......@@ -26,7 +26,7 @@ from utils.observation_utils import normalize_observation
checkpoint = "checkpoints/sample-checkpoint.pth"
# Beyond this number of agents, skip the episode
max_num_agents_handled = 79
max_env_width = 35
# Observation parameters
# These need to match your training parameters!
......@@ -108,7 +108,7 @@ while True:
obs_time, agent_time, step_time = 0.0, 0.0, 0.0
no_ops_mode = False
if nb_agents <= max_num_agents_handled and not check_if_all_blocked(env=local_env):
if local_env.width <= max_env_width and not check_if_all_blocked(env=local_env):
time_start = time.time()
action_dict = {}
for agent in range(nb_agents):
......
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