Reduce tutorial runtime

HumanCompatibleAI · Jul 6, 2023 · c303af1 · c303af1
1 parent 3c78336
commit c303af1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/tutorials/10_train_sqil.ipynb b/docs/tutorials/10_train_sqil.ipynb
@@ -46,7 +46,7 @@
     "    n_epochs=10,\n",
     "    n_steps=64,\n",
     ")\n",
-    "expert.learn(100_000)  # Note: set to 100000 to train a proficient expert"
+    "expert.learn(1_000)  # Note: set to 100_000 to train a proficient expert"
    ]
   },
   {
@@ -174,7 +174,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "sqil_trainer.train(total_timesteps=1_000_000)  # Note: set to 1_000_000 to obtain good results\n",
+    "sqil_trainer.train(total_timesteps=1_000)  # Note: set to 1_000_000 to obtain good results\n",
     "reward_after_training, _ = evaluate_policy(sqil_trainer.policy, env, 10)\n",
     "print(f\"Reward after training: {reward_after_training}\")"
    ]