diff options
| -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 f593c46..5dd4fb9 100644 --- a/models/test_model.py +++ b/models/test_model.py @@ -33,7 +33,7 @@ class TestModel(BaseModel): self.image_paths = input['A_paths'] def test(self): - self.real_A = Variable(self.input_A) + self.real_A = Variable(self.input_A, volatile=True) self.fake_B = self.netG(self.real_A) # get image paths |