From 250ce2c3cf80a6de52688d6a6d2bffe012bc047c Mon Sep 17 00:00:00 2001
From: "Harle, Antoine (Contracteur)" <Antoine.Harle@Teledyne.com>
Date: Fri, 31 Jan 2020 10:36:50 -0500
Subject: [PATCH] Enleve dependance RandAugment

---
 higher/smart_aug/datasets.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/higher/smart_aug/datasets.py b/higher/smart_aug/datasets.py
index 26a0975..d5a253c 100755
--- a/higher/smart_aug/datasets.py
+++ b/higher/smart_aug/datasets.py
@@ -30,13 +30,13 @@ transform = torchvision.transforms.Compose([
 #    torchvision.transforms.Normalize((0.4914, 0.4822, 0.4465), (0.2023, 0.1994, 0.2010)), #CIFAR10
 ])
 
-from RandAugment import RandAugment
-# Add RandAugment with N, M(hyperparameter)
 transform_train = torchvision.transforms.Compose([
     #transforms.RandomHorizontalFlip(),
     #transforms.RandomVerticalFlip(),
     torchvision.transforms.ToTensor(),
 ])
+#from RandAugment import RandAugment
+# Add RandAugment with N, M(hyperparameter)
 #transform_train.transforms.insert(0, RandAugment(n=2, m=30))
 
 ### Classic Dataset ###