From 079da5c02fd99ef35d7cad0e20c2924b7c2bcffd Mon Sep 17 00:00:00 2001
From: junyanz <junyanz@mit.edu>
Date: Sun, 18 Feb 2018 22:53:26 -0500
Subject: fix test_model & add timer for data loader & rename identity loss

---
 models/test_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'models/test_model.py')

diff --git a/models/test_model.py b/models/test_model.py
index 35b7402..f593c46 100644
--- a/models/test_model.py
+++ b/models/test_model.py
@@ -28,7 +28,7 @@ class TestModel(BaseModel):
         # we need to use single_dataset mode
         input_A = input['A']
         if len(self.gpu_ids) > 0:
-            input_A.cuda(self.gpu_ids[0], async=True)
+            input_A = input_A.cuda(self.gpu_ids[0], async=True)
         self.input_A = input_A
         self.image_paths = input['A_paths']
 
-- 
cgit v1.2.3-70-g09d2