Skip to content

Commit

Permalink
Merge pull request #590 from Farama-Foundation/dev
Browse files Browse the repository at this point in the history
Fix testing of trained policy in notebook
  • Loading branch information
eleurent authored Apr 14, 2024
2 parents dc5cdea + 413de13 commit c7ee099
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions scripts/intersection_social_dqn.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -147,8 +147,8 @@
"env = load_environment(env_config)\n",
"env.config[\"offscreen_rendering\"] = True\n",
"agent = load_agent(agent_config, env)\n",
"evaluation = Evaluation(env, agent, num_episodes=1)\n",
"evaluation.train()\n",
"evaluation = Evaluation(env, agent, num_episodes=20, training = False, recover = True)\n",
"evaluation.test()\n",
"show_videos(evaluation.run_directory)"
]
}
Expand Down

0 comments on commit c7ee099

Please sign in to comment.