wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix typo

This commit is contained in:
jayLEE0301 2024-06-04 16:53:49 -04:00
parent 6d9a65ca39
commit 0657c0c6c1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/policies/vqbet/modeling_vqbet.py
View File

@ -98,7 +98,7 @@ class VQBeTPolicy(nn.Module, PyTorchModelHubMixin):
# the dimension of returned action is (batch_size, n_action_pred_chunk, action_dim)
actions = self.unnormalize_outputs({"action": actions})["action"]
# since the data in the action queue's dimension is (n_action_pred_chunk, batch_size, action_dim, we transpose the action and fill the queue
# since the data in the action queue's dimension is (n_action_pred_chunk, batch_size, action_dim), we transpose the action and fill the queue
self._queues["action"].extend(actions.transpose(0, 1))
action = self._queues["action"].popleft()