Update keep_checkpoints

This commit is contained in:
Thomas Simonini
2022-12-15 09:53:37 +01:00
parent 05fe8dd33f
commit fa7eb7b0e4

View File

@@ -1,5 +1,15 @@
{
"cells": [
{
"cell_type": "markdown",
"metadata": {
"id": "view-in-github",
"colab_type": "text"
},
"source": [
"<a href=\"https://colab.research.google.com/github/huggingface/deep-rl-class/blob/main/notebooks/bonus-unit1/bonus-unit1.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
]
},
{
"cell_type": "markdown",
"metadata": {
@@ -320,7 +330,11 @@
"source": [
"- **In the case you want to modify the hyperparameters**, in Google Colab notebook, you can click here to open the config.yaml: `/content/ml-agents/config/ppo/Huggy.yaml`\n",
"\n",
"- For instance **if you want to save more models during the training** (for now, we save every 200,000 training timesteps). You need to modify:\n",
" - `checkpoint_interval`: The number of training timesteps collected between each checkpoint.\n",
" - `keep_checkpoints`: The maximum number of model checkpoints to keep. \n",
"\n",
"=> Just keep in mind that **decreasing the `checkpoint_interval` means more models to upload to the Hub and so a longer uploading time** \n",
"Were now ready to train our agent 🔥."
],
"metadata": {
@@ -541,7 +555,8 @@
"accelerator": "GPU",
"colab": {
"provenance": [],
"private_outputs": true
"private_outputs": true,
"include_colab_link": true
},
"gpuClass": "standard",
"kernelspec": {