Commit 9c240bcb authored by MasterScrat's avatar MasterScrat
Browse files

Using RL only for episodes with less than 80 agents

parent 6298ab02
......@@ -19,21 +19,21 @@ sys.path.append(str(base_dir))
from reinforcement_learning.dddqn_policy import DDDQNPolicy
from utils.observation_utils import normalize_observation
##########################
####################################################
# EVALUATION PARAMETERS #
# Checkpoint to use
checkpoint = "checkpoints/sample-checkpoint.pth"
# Beyond this number of agents, skip the episode
max_num_agents_handled = 0
max_num_agents_handled = 80
# Observation parameters
# These need to match your training parameters!
observation_tree_depth = 1
observation_radius = 10
observation_max_path_depth = 20
##########################
####################################################
remote_client = FlatlandRemoteClient()
......
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