diff --git a/notebooks/unit5/unit5.ipynb b/notebooks/unit5/unit5.ipynb
index fc163c0..cef401d 100644
--- a/notebooks/unit5/unit5.ipynb
+++ b/notebooks/unit5/unit5.ipynb
@@ -61,12 +61,12 @@
{
"cell_type": "markdown",
"source": [
- "### ๐ฎ Environments: \n",
+ "### ๐ฎ Environments:\n",
"\n",
"- [Pyramids](https://github.com/Unity-Technologies/ml-agents/blob/main/docs/Learning-Environment-Examples.md#pyramids)\n",
"- SnowballTarget\n",
"\n",
- "### ๐ RL-Library: \n",
+ "### ๐ RL-Library:\n",
"\n",
"- [ML-Agents](https://github.com/Unity-Technologies/ml-agents)\n"
],
@@ -117,7 +117,7 @@
"\n",
"- ๐ Study Deep Reinforcement Learning in **theory and practice**.\n",
"- ๐งโ๐ป Learn to **use famous Deep RL libraries** such as Stable Baselines3, RL Baselines3 Zoo, CleanRL and Sample Factory 2.0.\n",
- "- ๐ค Train **agents in unique environments** \n",
+ "- ๐ค Train **agents in unique environments**\n",
"\n",
"And more check ๐ the syllabus ๐ https://huggingface.co/deep-rl-course/communication/publishing-schedule\n",
"\n",
@@ -252,25 +252,23 @@
},
{
"cell_type": "markdown",
- "metadata": {
- "id": "jsoZGxr1MIXY"
- },
"source": [
- "Download the file SnowballTarget.zip from https://drive.google.com/file/d/1YHHLjyj6gaZ3Gemx1hQgqrPgSS2ZhmB5 using `wget`. \n",
- "\n",
- "Check out the full solution to download large files from GDrive [here](https://bcrf.biochem.wisc.edu/2021/02/05/download-google-drive-files-using-wget/)"
- ]
+ "We downloaded the file SnowballTarget.zip from https://github.com/huggingface/Snowball-Target using `wget`"
+ ],
+ "metadata": {
+ "id": "ekSh8LWawkB5"
+ }
},
{
"cell_type": "code",
- "execution_count": null,
- "metadata": {
- "id": "QU6gi8CmWhnA"
- },
- "outputs": [],
"source": [
- "!wget --load-cookies /tmp/cookies.txt \"https://docs.google.com/uc?export=download&confirm=$(wget --quiet --save-cookies /tmp/cookies.txt --keep-session-cookies --no-check-certificate 'https://docs.google.com/uc?export=download&id=1YHHLjyj6gaZ3Gemx1hQgqrPgSS2ZhmB5' -O- | sed -rn 's/.*confirm=([0-9A-Za-z_]+).*/\\1\\n/p')&id=1YHHLjyj6gaZ3Gemx1hQgqrPgSS2ZhmB5\" -O ./training-envs-executables/linux/SnowballTarget.zip && rm -rf /tmp/cookies.txt"
- ]
+ "!wget \"https://github.com/huggingface/Snowball-Target/raw/main/SnowballTarget.zip\" -O ./training-envs-executables/linux/SnowballTarget.zip"
+ ],
+ "metadata": {
+ "id": "6LosWO50wa77"
+ },
+ "execution_count": null,
+ "outputs": []
},
{
"cell_type": "markdown",
@@ -299,7 +297,7 @@
"id": "nyumV5XfPKzu"
},
"source": [
- "Make sure your file is accessible "
+ "Make sure your file is accessible"
]
},
{
@@ -400,9 +398,9 @@
"\n",
"![\"MlAgents](\"https://huggingface.co/datasets/huggingface-deep-rl-course/course-images/resolve/main/en/unit7/mlagentslearn.png\")
\n",
"\n",
- "Train the model and use the `--resume` flag to continue training in case of interruption. \n",
+ "Train the model and use the `--resume` flag to continue training in case of interruption.\n",
"\n",
- "> It will fail first time if and when you use `--resume`, try running the block again to bypass the error. \n",
+ "> It will fail first time if and when you use `--resume`, try running the block again to bypass the error.\n",
"\n"
]
},
@@ -449,7 +447,7 @@
"\n",
"![\"Create](\"https://huggingface.co/datasets/huggingface-deep-rl-course/course-images/resolve/main/en/notebooks/create-token.jpg\")
\n",
"\n",
- "- Copy the token \n",
+ "- Copy the token\n",
"- Run the cell below and paste the token"
],
"metadata": {
@@ -573,7 +571,7 @@
"1. In step 1, choose your model repository which is the model id (in my case ThomasSimonini/ppo-SnowballTarget).\n",
"\n",
"2. In step 2, **choose what model you want to replay**:\n",
- " - I have multiple one, since we saved a model every 500000 timesteps. \n",
+ " - I have multiple one, since we saved a model every 500000 timesteps.\n",
" - But if I want the more recent I choose `SnowballTarget.onnx`\n",
"\n",
"๐ Whatโs nice **is to try with different models step to see the improvement of the agent.**\n",
@@ -635,7 +633,7 @@
"id": "H7JmgOwcSSmF"
},
"source": [
- "Wait for the upload to finish and then run the command below. \n",
+ "Wait for the upload to finish and then run the command below.\n",
"\n",
""
]
@@ -667,7 +665,7 @@
"id": "KmKYBgHTAVMP"
},
"source": [
- "Make sure your file is accessible "
+ "Make sure your file is accessible"
]
},
{
@@ -769,7 +767,7 @@
"cell_type": "markdown",
"source": [
"### ๐ Bonus: Why not train on another environment?\n",
- "Now that you know how to train an agent using MLAgents, **why not try another environment?** \n",
+ "Now that you know how to train an agent using MLAgents, **why not try another environment?**\n",
"\n",
"MLAgents provides 17 different and weโre building some custom ones. The best way to learn is to try things of your own, have fun.\n",
"\n"
@@ -794,7 +792,7 @@
"\n",
"For the demos to visualize your agent ๐ https://huggingface.co/unity\n",
"\n",
- "For now we have integrated: \n",
+ "For now we have integrated:\n",
"- [Worm](https://huggingface.co/spaces/unity/ML-Agents-Worm) demo where you teach a **worm to crawl**.\n",
"- [Walker](https://huggingface.co/spaces/unity/ML-Agents-Walker) demo where you teach an agent **to walk towards a goal**."
],