From 4712e422bd5cfee7b5eac45a32b87a01c75b6ddd Mon Sep 17 00:00:00 2001 From: junyanz Date: Fri, 9 Feb 2018 18:07:05 -0500 Subject: cuda(...) inplace operation --- 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 f593c46..35b7402 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 = input_A.cuda(self.gpu_ids[0], async=True) + 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