fix modeling act

2024-05-29 10:04:27 +00:00
parent ad0033ae12
commit b24f2d0028
1 changed files with 1 additions and 1 deletions
--- a/lerobot/common/policies/act/modeling_act.py
+++ b/lerobot/common/policies/act/modeling_act.py
@@ -348,7 +348,7 @@ class ACT(nn.Module):
        images = batch["observation.images"]

        for cam_index in range(images.shape[-4]):
-            if self.config.n_obs_steps >= 1:
+            if self.config.n_obs_steps > 1:
                assert images.ndim == 6
                assert images.shape[1] == self.config.n_obs_steps
                cam_images = images[:, :, cam_index]