update demo

HumanCompatibleAI · Oct 6, 2023 · 125d19d · 125d19d
1 parent 68d1ac2
commit 125d19d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/train_dagger_atari_interactive_policy.py b/examples/train_dagger_atari_interactive_policy.py
@@ -56,7 +56,7 @@ def lr_schedule(_: float):
             rng=rng,
         )
         dagger_trainer.train(
-            total_timesteps=1,
+            total_timesteps=20,
             rollout_round_min_episodes=1,
-            rollout_round_min_timesteps=1,
+            rollout_round_min_timesteps=10,
         )