revert libero delta action change
This commit is contained in:
@@ -302,11 +302,13 @@ class LeRobotLiberoDataConfig(DataConfigFactory):
|
|||||||
# apply a separate delta conversion (that's why it's commented out). Choose whether to apply this
|
# apply a separate delta conversion (that's why it's commented out). Choose whether to apply this
|
||||||
# transform based on whether your dataset uses ``absolute`` or ``delta`` actions out of the box.
|
# transform based on whether your dataset uses ``absolute`` or ``delta`` actions out of the box.
|
||||||
|
|
||||||
# delta_action_mask = _transforms.make_bool_mask(6, -1)
|
# TODO(karl): comment this out once we have updated the Libero checkpoints to not use
|
||||||
# data_transforms = data_transforms.push(
|
# the delta action transform
|
||||||
# inputs=[_transforms.DeltaActions(delta_action_mask)],
|
delta_action_mask = _transforms.make_bool_mask(6, -1)
|
||||||
# outputs=[_transforms.AbsoluteActions(delta_action_mask)],
|
data_transforms = data_transforms.push(
|
||||||
# )
|
inputs=[_transforms.DeltaActions(delta_action_mask)],
|
||||||
|
outputs=[_transforms.AbsoluteActions(delta_action_mask)],
|
||||||
|
)
|
||||||
|
|
||||||
# Model transforms include things like tokenizing the prompt and action targets
|
# Model transforms include things like tokenizing the prompt and action targets
|
||||||
# You do not need to change anything here for your own dataset.
|
# You do not need to change anything here for your own dataset.
|
||||||
|
|||||||
Reference in New Issue
Block a user