learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-11-01 12:01:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

rl colab notebooks

Browse Source
This commit is contained in:
Varuna Jayasiri
2021-10-02 14:01:42 +05:30
parent 9e430d2dba
commit 8d1be06af5
2 changed files with 265 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
labml_nn/rl/ppo/experiment.ipynb
View File

@ -38,6 +38,25 @@
"!pip install labml-nn"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Add Atari ROMs (Doesn't work without this in Google Colab)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"! wget http://www.atarimania.com/roms/Roms.rar\n",
"! mkdir /content/ROM/\n",
"! unrar e /content/Roms.rar /content/ROM/\n",
"! python -m atari_py.import_roms /content/ROM/"
]
},
{
"cell_type": "markdown",
"metadata": {
@ -164,17 +183,7 @@
},
"outputs": [],
"source": [
"trainer = Trainer(\n",
" updates=configs['updates'],\n",
" epochs=configs['epochs'],\n",
" n_workers=configs['n_workers'],\n",
" worker_steps=configs['worker_steps'],\n",
" batches=configs['batches'],\n",
" value_loss_coef=configs['value_loss_coef'],\n",
" entropy_bonus_coef=configs['entropy_bonus_coef'],\n",
" clip_range=configs['clip_range'],\n",
" learning_rate=configs['learning_rate'],\n",
")"
"trainer = Trainer(**configs)"
]
},
{
@ -221,9 +230,9 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.8.5"
"version": "3.7.5"
}
},
"nbformat": 4,
"nbformat_minor": 4
}
}