diff options
| author | Boris Fomitchev <bfomitchev@nvidia.com> | 2018-05-08 00:56:35 -0700 |
|---|---|---|
| committer | Boris Fomitchev <bfomitchev@nvidia.com> | 2018-05-08 00:56:35 -0700 |
| commit | 4ca6b1610f9fa65f8bd7d7c15059bfde18a2f02a (patch) | |
| tree | ec2eeb09cdef6a70ea5612c3e6aa91ed2849414a /models/models.py | |
| parent | 736a2dc9afef418820e9c52f4f3b38460360b9f2 (diff) | |
Added data size and ONNX export options, FP16 inference is working
Diffstat (limited to 'models/models.py')
| -rwxr-xr-x | models/models.py | 3 |
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) |