You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
459 B
17 lines
459 B
### Copyright (C) 2020 Roy Or-El. All rights reserved.
|
|
### Licensed under the CC BY-NC-SA 4.0 license (https://creativecommons.org/licenses/by-nc-sa/4.0/legalcode).
|
|
import torch
|
|
|
|
def create_model(opt):
|
|
from .LATS_model import LATS, InferenceModel
|
|
if opt.isTrain:
|
|
model = LATS()
|
|
else:
|
|
model = InferenceModel()
|
|
|
|
model.initialize(opt)
|
|
if opt.verbose:
|
|
print("model [%s] was created" % (model.name()))
|
|
|
|
return model
|