Files
lerobot/tests
AdilZouitine 5ca3920611 feat(DeviceProcessor): Enhance tensor processing with device detection and float dtype conversion
- Improved the _process_tensor method to preserve GPU placement for tensors already on a GPU, facilitating multi-GPU training scenarios.
- Introduced a new _detect_device method in TokenizerProcessor to ensure tokenized tensors match the device of existing tensors in transitions.
- Added comprehensive unit tests to validate the functionality of device detection and float dtype conversion across various scenarios.
2025-08-08 19:33:24 +02:00
..
2025-07-01 16:34:46 +02:00
2025-03-13 14:05:55 +01:00
2025-07-01 16:34:46 +02:00
2025-07-01 16:34:46 +02:00
2025-07-01 16:34:46 +02:00
2025-07-16 11:31:25 +02:00
2025-07-01 16:34:46 +02:00