call self.reset() in ACT

2024-05-09 11:57:52 +01:00 · 2024-05-09 11:57:52 +01:00 · 2ea8ad4daf
parent 4cfebf1f0a
commit 2ea8ad4daf
1 changed files with 2 additions and 0 deletions
--- a/lerobot/common/policies/act/modeling_act.py
+++ b/lerobot/common/policies/act/modeling_act.py
@ -62,6 +62,8 @@ class ACTPolicy(nn.Module, PyTorchModelHubMixin):

        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: