From 891aec48124aacaf2128a65ff739804f645693b2 Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Thu, 16 Jan 2020 10:59:40 +0100
Subject: [PATCH] pin memory false by default

---
 pts/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pts/trainer.py b/pts/trainer.py
index 4f3ca71..c78b434 100644
--- a/pts/trainer.py
+++ b/pts/trainer.py
@@ -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,