Skip to content

Commit 808537c

Browse files
committed
Disable normalization
1 parent 7067e2d commit 808537c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/lerobot/policies/octo/modeling_octo.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -222,7 +222,8 @@ def _prepare_batch(
222222
Prepare batch for model input.
223223
Transforms a batch from the LeRobotDataset format to the format expected by the OctoModel.
224224
"""
225-
batch = self.normalize_inputs(batch)
225+
# TODO(lilkm): check normalization
226+
# batch = self.normalize_inputs(batch)
226227
# Get device from any available tensor in the batch
227228
device = next(iter(batch.values())).device
228229

0 commit comments

Comments
 (0)