Sign issue in modeling sac

2026-05-31 19:01:28 +00:00 · 2025-04-08 11:05:10 +02:00
parent 06de182448
commit 10adadbc71
3 changed files with 4 additions and 3 deletions
--- a/lerobot/scripts/server/buffer.py
+++ b/lerobot/scripts/server/buffer.py
@@ -79,7 +79,7 @@ def move_transition_to_device(transition: Transition, device: str = "cpu") -> Tr
                transition["complementary_info"][key] = val.to(device, non_blocking=non_blocking)
            elif isinstance(val, (int, float, bool)):
                transition["complementary_info"][key] = torch.tensor(
-                    val, device=device, non_blocking=non_blocking
+                    val, device=device
                )
            else:
                raise ValueError(f"Unsupported type {type(val)} for complementary_info[{key}]")