#2453 - Updates to Training-an-SB3-agent.ipynb following review

2024-04-18 10:23:26 +01:00
parent d05ded0dc2
commit 49be3b3639
1 changed files with 4 additions and 4 deletions
--- a/src/primaite/notebooks/Training-an-SB3-Agent.ipynb
+++ b/src/primaite/notebooks/Training-an-SB3-Agent.ipynb
@@ -48,9 +48,9 @@
    "from stable_baselines3 import PPO\n",
    "\n",
    "EPISODE_LEN = 128\n",
-    "NO_STEPS = EPISODE_LEN * 10\n",
-    "BATCH_SIZE = EPISODE_LEN * 10\n",
-    "TOTAL_TIMESTEPS = 5e3 * EPISODE_LEN\n",
+    "NUM_EPISODES = 10\n",
+    "NO_STEPS = EPISODE_LEN * NUM_EPISODES\n",
+    "BATCH_SIZE = 32\n",
    "LEARNING_RATE = 3e-4"
   ]
  },
@@ -69,7 +69,7 @@
   "metadata": {},
   "outputs": [],
   "source": [
-    "model.learn(total_timesteps=TOTAL_TIMESTEPS)\n"
+    "model.learn(total_timesteps=NO_STEPS)\n"
   ]
  },
  {