Spaces:
Runtime error
Runtime error
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 | |