Skip to content

Commit 92b1082

Browse files
authoredFeb 28, 2025
revert libero delta action change (Physical-Intelligence#351)
2 parents f1b9f4a + a4b1bf9 commit 92b1082

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed
 

‎src/openpi/training/config.py

+7-5
Original file line numberDiff line numberDiff line change
@@ -302,11 +302,13 @@ def create(self, assets_dirs: pathlib.Path, model_config: _model.BaseModelConfig
302302
# apply a separate delta conversion (that's why it's commented out). Choose whether to apply this
303303
# transform based on whether your dataset uses ``absolute`` or ``delta`` actions out of the box.
304304

305-
# delta_action_mask = _transforms.make_bool_mask(6, -1)
306-
# data_transforms = data_transforms.push(
307-
# inputs=[_transforms.DeltaActions(delta_action_mask)],
308-
# outputs=[_transforms.AbsoluteActions(delta_action_mask)],
309-
# )
305+
# TODO(karl): comment this out once we have updated the Libero checkpoints to not use
306+
# the delta action transform
307+
delta_action_mask = _transforms.make_bool_mask(6, -1)
308+
data_transforms = data_transforms.push(
309+
inputs=[_transforms.DeltaActions(delta_action_mask)],
310+
outputs=[_transforms.AbsoluteActions(delta_action_mask)],
311+
)
310312

311313
# Model transforms include things like tokenizing the prompt and action targets
312314
# You do not need to change anything here for your own dataset.

0 commit comments

Comments
 (0)
Please sign in to comment.