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

..

test_batch_conversion.py

refactor(pipeline): Transition from tuple to dictionary format for EnvTransition

2025-08-01 08:41:53 +02:00

test_batch_processor.py

feature(pipeline): port tokenizer pipeline for VLA (#1645)

2025-08-06 17:21:13 +02:00

test_converters.py

Integrate pipeline and add phone teleop (#1681)

2025-08-07 16:13:34 +02:00

test_device_processor.py

feat(DeviceProcessor): Enhance tensor processing with device detection and float dtype conversion

2025-08-08 19:33:24 +02:00

test_normalize_processor.py

refactor(normalization): remove Normalize and Unnormalize classes

2025-08-08 13:23:10 +02:00

test_observation_processor.py

Integrate pipeline and add phone teleop (#1681)

2025-08-07 16:13:34 +02:00

test_pipeline.py

Integrate pipeline and add phone teleop (#1681)

2025-08-07 16:13:34 +02:00

test_rename_processor.py

Integrate pipeline and add phone teleop (#1681)

2025-08-07 16:13:34 +02:00

test_tokenizer_processor.py

feat(DeviceProcessor): Enhance tensor processing with device detection and float dtype conversion

2025-08-08 19:33:24 +02:00