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

Remove batch length restrictions in select_action (#123)

Browse Source
This commit is contained in:
Simon Alibert
2024-05-04 15:33:55 +02:00
committed by GitHub
parent bccee745c3
commit c015252e20
3 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/common/policies/diffusion/modeling_diffusion.py
View File

@@ -115,7 +115,6 @@ class DiffusionPolicy(nn.Module, PyTorchModelHubMixin):
"""
assert "observation.image" in batch
assert "observation.state" in batch
assert len(batch) == 2
batch = self.normalize_inputs(batch)