From 413de131635be3b46a1378b8d0d9b07e8c8e11a2 Mon Sep 17 00:00:00 2001 From: Edouard Leurent Date: Sun, 14 Apr 2024 11:30:38 +0100 Subject: [PATCH] Fix testing of trained policy in notebook see https://github.com/Farama-Foundation/HighwayEnv/issues/586 --- scripts/intersection_social_dqn.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/intersection_social_dqn.ipynb b/scripts/intersection_social_dqn.ipynb index 9cef8b0d4..e29f32acb 100644 --- a/scripts/intersection_social_dqn.ipynb +++ b/scripts/intersection_social_dqn.ipynb @@ -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)" ] }