oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Charlie Crane
2024-04-18 10:23:26 +01:00
parent d05ded0dc2
commit 49be3b3639
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/primaite/notebooks/Training-an-SB3-Agent.ipynb
View File

@@ -48,9 +48,9 @@
"from stable_baselines3 import PPO\n", "from stable_baselines3 import PPO\n",
"\n", "\n",
"EPISODE_LEN = 128\n", "EPISODE_LEN = 128\n",
"NO_STEPS = EPISODE_LEN * 10\n", "NUM_EPISODES = 10\n",
"BATCH_SIZE = EPISODE_LEN * 10\n", "NO_STEPS = EPISODE_LEN * NUM_EPISODES\n",
"TOTAL_TIMESTEPS = 5e3 * EPISODE_LEN\n", "BATCH_SIZE = 32\n",
"LEARNING_RATE = 3e-4" "LEARNING_RATE = 3e-4"
] ]
}, },
@@ -69,7 +69,7 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"model.learn(total_timesteps=TOTAL_TIMESTEPS)\n" "model.learn(total_timesteps=NO_STEPS)\n"
] ]
}, },
{ {