diff options
Diffstat (limited to 'models')
| -rw-r--r-- | models/__init__.py | 2 | ||||
| -rw-r--r-- | models/test_model.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/models/__init__.py b/models/__init__.py index 681c6de..72a0d2e 100644 --- a/models/__init__.py +++ b/models/__init__.py @@ -10,7 +10,7 @@ def create_model(opt): from .pix2pix_model import Pix2PixModel model = Pix2PixModel() elif opt.model == 'test': - assert(opt.dataset_mode == 'single') + assert(opt.dataset_mode == 'single' or opt.dataset_mode == 'recursive') from .test_model import TestModel model = TestModel() else: 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 |