Make environment reset reinstantiate the game

2024-02-20 16:21:03 +00:00
parent 88f8e9cb42
commit 64b9ba3ecf
3 changed files with 21 additions and 15 deletions
--- a/src/primaite/notebooks/training_example_sb3.ipynb
+++ b/src/primaite/notebooks/training_example_sb3.ipynb
@@ -38,7 +38,7 @@
   "metadata": {},
   "outputs": [],
   "source": [
-    "gym = PrimaiteGymEnv(game=game)"
+    "gym = PrimaiteGymEnv(game_config=cfg)"
   ]
  },
  {
@@ -65,7 +65,7 @@
   "metadata": {},
   "outputs": [],
   "source": [
-    "model.learn(total_timesteps=1000)\n"
+    "model.learn(total_timesteps=10)\n"
   ]
  },
  {