Update DataLoader.py

vita-epfl · Nov 16, 2020 · 2d35396 · 2d35396
1 parent ef751ac
commit 2d35396
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/DataLoader.py b/DataLoader.py
@@ -10,7 +10,6 @@
 import cv2
 import matplotlib.pyplot as plt
 import time
-import import_ipynb
 import utils
 import matplotlib.pyplot as plt
 
@@ -224,4 +223,4 @@ def data_loader(args):
             test_set, batch_size=args.batch_size, shuffle=args.loader_shuffle,
             pin_memory=args.pin_memory, num_workers=args.loader_workers, drop_last=True)
 
-        return test_loader
+        return test_loader