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.
35 lines
1.1 KiB
35 lines
1.1 KiB
"""This package contains modules related to function, network architectures, and models"""
|
|
|
|
import importlib
|
|
from .base_model import BaseModel
|
|
|
|
|
|
def find_model_using_name(model_name):
|
|
"""Import the module "model/[model_name]_model.py"."""
|
|
model_file_name = "model." + model_name + "_model"
|
|
modellib = importlib.import_module(model_file_name)
|
|
model = None
|
|
for name, cls in modellib.__dict__.items():
|
|
if name.lower() == model_name.lower() and issubclass(cls, BaseModel):
|
|
model = cls
|
|
|
|
if model is None:
|
|
print("In %s.py, there should be a subclass of BaseModel with class name that matches %s in lowercase." % (model_file_name, model_name))
|
|
exit(0)
|
|
|
|
return model
|
|
|
|
|
|
def get_option_setter(model_name):
|
|
"""Return the static method <modify_commandline_options> of the model class."""
|
|
model = find_model_using_name(model_name)
|
|
return model.modify_options
|
|
|
|
|
|
def create_model(opt):
|
|
"""Create a model given the option."""
|
|
model = find_model_using_name(opt.model)
|
|
instance = model(opt)
|
|
print("model [%s] was created" % type(instance).__name__)
|
|
return instance
|