Skip to content

Commit

Permalink
Unit3: Removed order of magnitude for timesteps and a duplicated ffmp…
Browse files Browse the repository at this point in the history
…eg installation

1. With `n_timesteps: !!float 1e7` it takes a whole day to train a model. As it was suggested [in the discord channel](https://discord.com/channels/879548962464493619/971379033642266654/1113757769049321472) change number of timesteps to 1e7
2. `ffmpeg` is being installed twice. Removed 2nd attempt.
  • Loading branch information
medvedev authored Oct 20, 2023
1 parent df40414 commit 3964f50
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions notebooks/unit3/unit3.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -277,7 +277,6 @@
"source": [
"%%capture\n",
"!apt install python-opengl\n",
"!apt install ffmpeg\n",
"!apt install xvfb\n",
"!pip3 install pyvirtualdisplay"
]
Expand Down Expand Up @@ -317,7 +316,7 @@
" - stable_baselines3.common.atari_wrappers.AtariWrapper\n",
" frame_stack: 4\n",
" policy: 'CnnPolicy'\n",
" n_timesteps: !!float 1e7\n",
" n_timesteps: !!float 1e6\n",
" buffer_size: 100000\n",
" learning_rate: !!float 1e-4\n",
" batch_size: 32\n",
Expand Down Expand Up @@ -830,4 +829,4 @@
},
"nbformat": 4,
"nbformat_minor": 0
}
}

0 comments on commit 3964f50

Please sign in to comment.