call self.reset() in ACT

huggingface · May 9, 2024 · 2ea8ad4 · 2ea8ad4
1 parent 4cfebf1
commit 2ea8ad4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lerobot/common/policies/act/modeling_act.py b/lerobot/common/policies/act/modeling_act.py
@@ -62,6 +62,8 @@ def __init__(
 
         self.expected_image_keys = [k for k in config.input_shapes if k.startswith("observation.image")]
 
+        self.reset()
+
     def reset(self):
         """This should be called whenever the environment is reset."""
         if self.config.n_action_steps is not None: