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

check output dir whether empty

Browse Source
This commit is contained in:
Ray Wong 2020-08-08 07:09:56 +08:00
parent 206d9343cd
commit 888a052f05
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
main.py
View File

@ -29,14 +29,16 @@ def running(local_rank, config, task, backup_config=False, setup_output_dir=Fals
if setup_output_dir: if setup_output_dir:
output_dir = Path(config.result_dir) / config.name if config.output_dir is None else config.output_dir output_dir = Path(config.result_dir) / config.name if config.output_dir is None else config.output_dir
config.output_dir = str(output_dir) config.output_dir = str(output_dir)
if idist.get_rank() == 0: if output_dir.exists():
if not output_dir.exists(): assert not any(output_dir.iterdir()), "output_dir must be empty"
else:
if idist.get_rank() == 0:
output_dir.mkdir(parents=True) output_dir.mkdir(parents=True)
logger.info(f"mkdir -p {output_dir}") logger.info(f"mkdir -p {output_dir}")
logger.info(f"output path: {config.output_dir}") logger.info(f"output path: {config.output_dir}")
if backup_config: if backup_config and idist.get_rank() == 0:
with open(output_dir / "config.yml", "w+") as f: with open(output_dir / "config.yml", "w+") as f:
print(config.pretty(), file=f) print(config.pretty(), file=f)
OmegaConf.set_readonly(config, True) OmegaConf.set_readonly(config, True)