Skip to content

Commit

Permalink
Remove cell metadata from GAIL tutorial.
Browse files Browse the repository at this point in the history
  • Loading branch information
ernestum committed Dec 8, 2023
1 parent 8c6ff8e commit ed13f42
Showing 1 changed file with 12 additions and 24 deletions.
36 changes: 12 additions & 24 deletions docs/tutorials/3_train_gail.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -124,69 +124,57 @@
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"... then we evaluate it bofere training ..."
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"env.seed(SEED)\n",
"learner_rewards_before_training, _ = evaluate_policy(\n",
" learner, env, 100, return_episode_rewards=True\n",
")"
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"... and train it ..."
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"gail_trainer.train(200_000)"
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"... and finally evaluate it again."
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"env.seed(SEED)\n",
"learner_rewards_after_training, _ = evaluate_policy(\n",
" learner, env, 100, return_episode_rewards=True\n",
")"
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "markdown",
Expand Down

0 comments on commit ed13f42

Please sign in to comment.