From e3726af25b83134d6240b926386fa0243f6a6a96 Mon Sep 17 00:00:00 2001 From: junyanz Date: Sun, 15 Apr 2018 00:51:38 -0400 Subject: fix memory issue in the test_model.py --- 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..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 -- cgit v1.2.3-70-g09d2