update ml-starter

codekansas · Nov 2, 2023 · ecb1c33 · ecb1c33
1 parent 95c28a1
commit ecb1c33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/project/tasks/rl_demo.py b/project/tasks/rl_demo.py
@@ -134,7 +134,7 @@ def compute_loss(self, model: A2CModel, batch: tuple[State, Action], state: ml.S
         # Logs additional metrics.
         self.logger.log_scalar("p_dist_std", lambda: p_dist.stddev.mean().item())
         if state.num_epoch_steps == 0 and state.num_epochs % self.config.sample_clip_interval == 0:
-            self.logger.log_video("sample", self.sample_clip(model=model, use_tqdm=False))
+            self.logger.log_video("sample", self.sample_clip(model=model))
 
         return {
             "policy": policy_loss,