diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-06-13 00:10:04 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-06-13 00:10:04 -0700 |
| commit | 7212499f839db946c9296b84d6491fe5d6097dea (patch) | |
| tree | 95f05a1163d09ce9e56bde60f6b332bd529e5f5c /models/models.py | |
| parent | e6858e35f0a08c6139c133122d222d0d85e8005d (diff) | |
predict single image
Diffstat (limited to 'models/models.py')
| -rw-r--r-- | models/models.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/models.py b/models/models.py index efcd898..d5bb9d8 100644 --- a/models/models.py +++ b/models/models.py @@ -13,7 +13,7 @@ def create_model(opt): elif opt.model == 'test': assert(opt.dataset_mode == 'single') from .test_model import TestModel - model = OneDirectionTestModel() + model = TestModel() else: raise ValueError("Model [%s] not recognized." % opt.model) model.initialize(opt) |