refactor(train): Remove unnecessary tensor device handling in training loop

2026-05-19 10:40:04 +00:00 · 2025-08-08 19:35:15 +02:00
parent 5bf82f8229
commit fd5d8b3d5f
1 changed files with 0 additions and 4 deletions
@@ -209,10 +209,6 @@ def train(cfg: TrainPipelineConfig):
        batch = preprocessor(batch)
        train_tracker.dataloading_s = time.perf_counter() - start_time

-        for key in batch:
-            if isinstance(batch[key], torch.Tensor):
-                batch[key] = batch[key].to(device, non_blocking=device.type == "cuda")
-
        train_tracker, output_dict = update_policy(
            train_tracker,
            policy,