CS_ALearn/deep-rl-class
1
0
Fork 0
mirror of https://github.com/huggingface/deep-rl-class.git synced 2026-04-03 10:38:27 +08:00
Code Issues Projects Releases Wiki Activity

Add step to specify model file name

Browse Source
This commit is contained in:
Daisuke Shimamoto
2022-05-06 00:34:16 +09:00
parent bf587c19b6
commit e7611ec600
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
unit1/unit1.ipynb
View File

@@ -695,7 +695,10 @@
},
"outputs": [],
"source": [
"# Train it for 500,000 timesteps"
"# TODO: Train it for 500,000 timesteps\n",
"\n",
"# TODO: Specify file name for model and save the model to file\n",
"model_name = \"\"\n"
]
},
{
@@ -717,7 +720,10 @@
"source": [
"# SOLUTION\n",
"# Train it for 500,000 timesteps\n",
"model.learn(total_timesteps=500000)"
"model.learn(total_timesteps=500000)\n",
"# Save the model\n",
"model_name = \"ppo-LunarLander-v2\"\n",
"model.save(model_name)"
]
},
{
@@ -980,9 +986,6 @@
"# Define the name of the environment\n",
"env_id = \"LunarLander-v2\"\n",
"\n",
"# Define the name of the trained model that we defined in model_save\n",
"model_name = \"ppo-LunarLander-v2\"\n",
"\n",
"# TODO: Define the model architecture we used\n",
"model_architecture = \"PPO\"\n",
"\n",