summaryrefslogtreecommitdiff
path: root/models
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-12 17:12:17 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-12 17:12:17 +0200
commitef994e09872bcc3b4b34bd63d1e8302fff2bc0b2 (patch)
tree950d7d45bd352ecdce887a2db7fafe4b105c9644 /models
parent5811bcaf522a0a1c95861b4222c8a9a8da3ff358 (diff)
spaces... wtf
Diffstat (limited to 'models')
-rwxr-xr-xmodels/models.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/models/models.py b/models/models.py
index 8e72e46..f5bc8d0 100755
--- a/models/models.py
+++ b/models/models.py
@@ -4,14 +4,14 @@ import torch
def create_model(opt):
if opt.model == 'pix2pixHD':
- from .pix2pixHD_model import Pix2PixHDModel, InferenceModel
- if opt.isTrain:
+ from .pix2pixHD_model import Pix2PixHDModel, InferenceModel
+ if opt.isTrain:
model = Pix2PixHDModel()
else:
model = InferenceModel()
else:
- from .ui_model import UIModel
- model = UIModel()
+ from .ui_model import UIModel
+ model = UIModel()
model.initialize(opt)
if opt.verbose:
print("model [%s] was created" % (model.name()))