Spaces:
Runtime error
Runtime error
File size: 1,686 Bytes
bc32eea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
print("HIIIIIIOOO")
import importlib
from .base_model import BaseModel
print("HIIIIIIOOO")
def find_model_using_name(model_name):
# Given the option --model [modelname],
# the file "models/modelname_model.py"
# will be imported.
# task_module = importlib.import_module(task_name)
# model_filename = task_name + ".models." + model_name.lower() + "_model"
model_filename = "models." + model_name.lower() + "_model"
# modellib = importlib.import_module(model_filename, package=task_module)
modellib = importlib.import_module(model_filename)
# In the file, the class called ModelNameModel() will
# be instantiated. It has to be a subclass of BaseModel,
# and it is case-insensitive.
model = None
target_model_name = model_name.replace('_', '') + 'model'
for name, cls in modellib.__dict__.items():
if name.lower() == target_model_name.lower() \
and next(iter(cls.__bases__)).__module__.endswith(BaseModel.__module__): # check that base class is BaseModel
model = cls
if model is None:
raise NotImplementedError("In %s.py, there should be a subclass of BaseModel with class name that matches %s in lowercase." % (model_filename, target_model_name))
return model
def get_option_setter(model_name):
model_class = find_model_using_name(model_name)
return model_class.modify_commandline_options
def create_model(opt):
instance = create_model_by_name(opt.model, opt)
return instance
def create_model_by_name(name, opt):
model = find_model_using_name(name)
instance = model(opt)
print("model [%s] was created" % (instance.name()))
return instance
|