pin memory false by default

2026-08-01 12:51:16 +08:00 · 2020-01-16 10:59:40 +01:00
parent 0b1fa27f55
commit 891aec4812
1 changed files with 1 additions and 1 deletions
@@ -14,7 +14,7 @@ class Trainer:
        batch_size: int = 32,
        num_batches_per_epoch: int = 50,
        num_workers: int = 4,
-        pin_memory: bool = True,
+        pin_memory: bool = False,
        learning_rate: float = 1e-3,
        weight_decay: float = 1e-6,
        device: Optional[torch.device] = None,