From 7212499f839db946c9296b84d6491fe5d6097dea Mon Sep 17 00:00:00 2001 From: junyanz Date: Tue, 13 Jun 2017 00:10:04 -0700 Subject: predict single image --- models/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'models/models.py') 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) -- cgit v1.2.3-70-g09d2