From b67ec3c46994341aad1cbdb5d7f6aa2a98be4051 Mon Sep 17 00:00:00 2001
From: "Harle, Antoine (Contracteur)" <Antoine.Harle@Teledyne.com>
Date: Fri, 17 Jan 2020 15:12:28 -0500
Subject: [PATCH] Resolution pb memoire apres 1 epoch (Batch size test)

---
 higher/datasets.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/higher/datasets.py b/higher/datasets.py
index f6fe438..9d669cc 100755
--- a/higher/datasets.py
+++ b/higher/datasets.py
@@ -3,8 +3,8 @@ from torch.utils.data import SubsetRandomSampler
 import torchvision
 
 BATCH_SIZE = 300
-#TEST_SIZE = 300
-TEST_SIZE = 10000
+TEST_SIZE = 300 
+#TEST_SIZE = 10000 #legerement +Rapide / + Consomation memoire !
 
 #ATTENTION : Dataug (Kornia) Expect image in the range of [0, 1]
 #transform_train = torchvision.transforms.Compose([
@@ -35,7 +35,7 @@ import augmentation_transforms
 import numpy as np
 
 download_data=False
-num_workers=0
+num_workers=4 #16
 pin_memory=False
 
 class AugmentedDataset(VisionDataset):