admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-25 13:40:00 +00:00
Code Issues 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
lerobot/common/policies/diffusion/modeling_diffusion.py
-1
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)