summaryrefslogtreecommitdiff
path: root/models/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'models/models.py')
-rwxr-xr-xmodels/models.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/models/models.py b/models/models.py
index 0ba442f..805696f 100755
--- a/models/models.py
+++ b/models/models.py
@@ -10,7 +10,8 @@ def create_model(opt):
from .ui_model import UIModel
model = UIModel()
model.initialize(opt)
- print("model [%s] was created" % (model.name()))
+ if opt.verbose:
+ print("model [%s] was created" % (model.name()))
if opt.isTrain and len(opt.gpu_ids):
model = torch.nn.DataParallel(model, device_ids=opt.gpu_ids)