9 lines
322 B
Python
9 lines
322 B
Python
from omegaconf import OmegaConf
|
|
from ..config import instantiate_from_config
|
|
|
|
def build_model(cfg, datamodule):
|
|
model_config = OmegaConf.to_container(cfg.model, resolve=True)
|
|
model_config['params']['cfg'] = cfg
|
|
model_config['params']['datamodule'] = datamodule
|
|
return instantiate_from_config(model_config)
|