Fix testing of trained policy in notebook

see #586
Farama-Foundation · Apr 14, 2024 · 413de13 · 413de13
1 parent dc5cdea
commit 413de13
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)"
       ]
     }