typo fix

Author: avaamini

lab3/solutions/RL_Solution.ipynb

@@ -5,7 +5,9 @@
     "colab": {
       "name": "RL_Solution.ipynb",
       "provenance": [],
-      "collapsed_sections": []
+      "collapsed_sections": [
+        "jrI6q7RmWQam"
+      ]
     },
     "kernelspec": {
       "name": "python3",
@@ -960,7 +962,7 @@
         "# Rollout with single batch\r\n",
         "single_batch_size = 1\r\n",
         "memories = collect_rollout(single_batch_size, env, test_model, choose_action)\r\n",
-        "rollout_video = mdl.lab3.save_video_of_memory(memories[0], \"CartPole-Random-Agent.mp4\")\r\n",
+        "rollout_video = mdl.lab3.save_video_of_memory(memories[0], \"Pong-Random-Agent.mp4\")\r\n",
         "\r\n",
         "# Play back video of memories\r\n",
         "mdl.lab3.play_video(rollout_video)"
@@ -1093,8 +1095,8 @@
       },
       "source": [
         "latest_pong = mdl.lab3.save_video_of_model(\n",
-        "    pong_model, \"Pong-v0\", suffix=\"latest\")\n",
-        "mdl.lab3.play_video(latest_pong)"
+        "    pong_model, \"Pong-v0\", suffix=\"_latest\")\n",
+        "mdl.lab3.play_video(latest_pong, width=400)"
       ],
       "execution_count": null,
       "outputs": []