From 016b50b4da2ac2457b8de033c7329406db6fcef1 Mon Sep 17 00:00:00 2001 From: Yachen Kang Date: Wed, 26 Feb 2025 15:06:21 +0800 Subject: [PATCH] change "actions_id_pad" to "actions_is_pad" Is this a typo or intentional? --- lerobot/common/policies/pi0/modeling_pi0.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/policies/pi0/modeling_pi0.py b/lerobot/common/policies/pi0/modeling_pi0.py index c8b12cafcd..bc53bf85bb 100644 --- a/lerobot/common/policies/pi0/modeling_pi0.py +++ b/lerobot/common/policies/pi0/modeling_pi0.py @@ -313,7 +313,7 @@ def forward(self, batch: dict[str, Tensor], noise=None, time=None) -> tuple[Tens state = self.prepare_state(batch) lang_tokens, lang_masks = self.prepare_language(batch) actions = self.prepare_action(batch) - actions_is_pad = batch.get("actions_id_pad") + actions_is_pad = batch.get("actions_is_pad") loss_dict = {} losses = self.model.forward(images, img_masks, lang_tokens, lang_masks, state, actions, noise, time)