OpenMotionLab/MotionGPT

1from omegaconf import OmegaConf2from os.path import join as pjoin3from mGPT.config import instantiate_from_config4 5 6def build_data(cfg, phase="train"):7    data_config = OmegaConf.to_container(cfg.DATASET, resolve=True)8    data_config['params'] = {'cfg': cfg, 'phase': phase}9    if isinstance(data_config['target'], str):10        return instantiate_from_config(data_config)11    elif isinstance(data_config['target'], list):12        data_config_tmp = data_config.copy()13        data_config_tmp['params']['dataModules'] = data_config['target']14        data_config_tmp['target'] = 'mGPT.data.Concat.ConcatDataModule'15        return instantiate_from_config(data_config)16