diff options
| author | junyanz <junyanz@mit.edu> | 2018-02-18 22:53:26 -0500 |
|---|---|---|
| committer | junyanz <junyanz@mit.edu> | 2018-02-18 22:53:26 -0500 |
| commit | 079da5c02fd99ef35d7cad0e20c2924b7c2bcffd (patch) | |
| tree | 8d00061c5b639f0ad832ab2a34a76d10a076dff7 /models/test_model.py | |
| parent | 51bd910bc737984163f6d6534cccbd22668e2b28 (diff) | |
fix test_model & add timer for data loader & rename identity loss
Diffstat (limited to 'models/test_model.py')
| -rw-r--r-- | models/test_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
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'] |