diff --git a/main.py b/main.py index 1776f4b..2edb96e 100644 --- a/main.py +++ b/main.py @@ -48,7 +48,10 @@ def running(local_rank, config, task, backup_config=False, setup_output_dir=Fals def run(task, config: str, *omega_options, **kwargs): omega_options = [str(o) for o in omega_options] - conf = OmegaConf.merge(OmegaConf.load(config), OmegaConf.from_cli(omega_options)) + cli_conf = OmegaConf.from_cli(omega_options) + if len(cli_conf) > 0: + print(cli_conf.pretty()) + conf = OmegaConf.merge(OmegaConf.load(config), cli_conf) backend = kwargs.get("backend", "nccl") backup_config = kwargs.get("backup_config", False) setup_output_dir = kwargs.get("setup_output_dir", False)