File size: 325 Bytes
4409449 |
1 2 3 4 5 6 7 8 9 |
from omegaconf import OmegaConf
from mGPT.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)
|