diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-04-18 12:35:39 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-04-18 12:35:39 +0200 |
| commit | d3044d50514586b0cf9702bc2c16a6486af249f6 (patch) | |
| tree | 2c05c6d37efe7b23abc93c5419d6b2cfb037e6bb /models | |
| parent | e3726af25b83134d6240b926386fa0243f6a6a96 (diff) | |
setting up recursive dataset
Diffstat (limited to 'models')
| -rw-r--r-- | models/__init__.py | 2 |
1 files changed, 1 insertions, 1 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: |