diff options
| author | junyanz <junyanzhu89@gmail.com> | 2018-04-15 00:51:38 -0400 |
|---|---|---|
| committer | junyanz <junyanzhu89@gmail.com> | 2018-04-15 00:51:38 -0400 |
| commit | e3726af25b83134d6240b926386fa0243f6a6a96 (patch) | |
| tree | 0c09b77886332f7a3711c259514da84286bbd0e7 | |
| parent | 1cf7171641711d69663733989c267110d66ca466 (diff) | |
fix memory issue in the 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 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 |