tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix modeling act

Browse Source
This commit is contained in:
Remi Cadene
2024-05-29 10:04:27 +00:00
parent ad0033ae12
commit b24f2d0028
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/policies/act/modeling_act.py
View File

@@ -348,7 +348,7 @@ class ACT(nn.Module):
images = batch["observation.images"] images = batch["observation.images"]
for cam_index in range(images.shape[-4]): 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.ndim == 6
assert images.shape[1] == self.config.n_obs_steps assert images.shape[1] == self.config.n_obs_steps
cam_images = images[:, :, cam_index] cam_images = images[:, :, cam_index]