Reduce the number of train steps in the FAST mode of the AIRL tutoria…

…l notebook.
HumanCompatibleAI · Oct 9, 2023 · 0ed0d17 · 0ed0d17
1 parent edeb9f0
commit 0ed0d17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/tutorials/4_train_airl.ipynb b/docs/tutorials/4_train_airl.ipynb
@@ -33,7 +33,7 @@
     "FAST = True\n",
     "\n",
     "if FAST:\n",
-    "    N_RL_TRAIN_STEPS = 300_000\n",
+    "    N_RL_TRAIN_STEPS = 100_000\n",
     "else:\n",
     "    N_RL_TRAIN_STEPS = 2_000_000\n",
     "\n",