Skip to content
Snippets Groups Projects
Commit 6fb987ed authored by Dipam Chakraborty's avatar Dipam Chakraborty
Browse files

Update agent.py

parent 92d19411
No related branches found
No related tags found
No related merge requests found
...@@ -84,7 +84,8 @@ if __name__ == '__main__': ...@@ -84,7 +84,8 @@ if __name__ == '__main__':
infos = [{} for _ in range(num_envs)] infos = [{} for _ in range(num_envs)]
# Change this to your agent interface # Change this to your agent interface
num_actions = batched_env.envs[0].action_space.n # num_actions = batched_env.envs[0].action_space.n
num_actions = 112
agent = RandomBatchedAgent(num_envs, num_actions) agent = RandomBatchedAgent(num_envs, num_actions)
# The evaluation setup will automatically stop after the requisite number of rollouts # The evaluation setup will automatically stop after the requisite number of rollouts
......
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