admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 03:59:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

split up select action return statement

Browse Source
This commit is contained in:
Bryson Jones
2025-12-10 13:50:47 -08:00
parent 3a16a002f8
commit 5524a0d7a7
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py
+2 -1
View File
@@ -795,7 +795,8 @@ class MultiTaskDiTPolicy(PreTrainedPolicy):
actions = self.predict_action_chunk(batch) actions = self.predict_action_chunk(batch)
self._queues[ACTION].extend(actions.transpose(0, 1)) self._queues[ACTION].extend(actions.transpose(0, 1))
return self._queues[ACTION].popleft() action = self._queues[ACTION].popleft()
return action
def forward(self, batch: dict[str, Tensor]) -> tuple[Tensor, dict | None]: def forward(self, batch: dict[str, Tensor]) -> tuple[Tensor, dict | None]:
"""Run the batch through the model and compute the loss for training""" """Run the batch through the model and compute the loss for training"""