budui/raycv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

show cli config when it is not empty

Browse Source
This commit is contained in:
Ray Wong 2020-08-08 07:34:42 +08:00
parent 8abd35467c
commit c520ce9501
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
main.py
View File

@ -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): def run(task, config: str, *omega_options, **kwargs):
omega_options = [str(o) for o in omega_options] 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") backend = kwargs.get("backend", "nccl")
backup_config = kwargs.get("backup_config", False) backup_config = kwargs.get("backup_config", False)
setup_output_dir = kwargs.get("setup_output_dir", False) setup_output_dir = kwargs.get("setup_output_dir", False)