from local

This commit is contained in:
dohe0342 2023-01-08 19:03:57 +09:00
parent 839b8c0859
commit 7b2a0fdb19
3 changed files with 3 additions and 0 deletions

View File

@ -1186,6 +1186,9 @@ def main():
args = parser.parse_args()
args.exp_dir = Path(args.exp_dir)
logging.info("save arguments to config.yaml...")
save_args(args)
world_size = args.world_size
assert world_size >= 1
if world_size > 1: