bf

pytorch · vmoens · May 7, 2022 · Apr 28, 2022 · Apr 28, 2022 · Apr 28, 2022
commit 106b601ca43bc27417e020d564ff62c89c26475a
diff --git a/torchrl/trainers/trainers.py b/torchrl/trainers/trainers.py
@@ -794,7 +794,7 @@ def __call__(self, batch: _TensorDict) -> Tuple[str, int]:
         else:
             current_frames = batch.numel() * self.frame_skip
         self.frame_count += current_frames
-        return "collected_frames", self.frame_count
+        return "n_frames", self.frame_count
 
 
 def _check_input_output_typehint(func: Callable, input: Type, output: Type):