From 7f8b0de93cf8a12d5408db200a0ad2459cb0fee5 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 18 Apr 2018 13:20:56 +0200
Subject: k

---
 test.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'test.py')

diff --git a/test.py b/test.py
index 78d0c98..1ddb894 100644
--- a/test.py
+++ b/test.py
@@ -21,7 +21,8 @@ if __name__ == '__main__':
     web_dir = os.path.join(opt.results_dir, opt.name, '%s_%s' % (opt.phase, opt.which_epoch))
     webpage = html.HTML(web_dir, 'Experiment = %s, Phase = %s, Epoch = %s' % (opt.name, opt.phase, opt.which_epoch))
     # test
-    for i, data in enumerate(dataset):
+    print(dataset.name())
+    for i, data in enumerate(data_loader):
         if i >= opt.how_many:
             break
         model.set_input(data)
-- 
cgit v1.2.3-70-g09d2