from local

This commit is contained in:
dohe0342 2022-12-26 13:29:33 +09:00
parent 5a4ea47d0d
commit 312622fc46
2 changed files with 4 additions and 4 deletions

Binary file not shown.

View File

@ -1419,10 +1419,10 @@ def is_module_available(*modules: str) -> bool:
def save_args(args):
if not os.path.exists(args.exp_dir):
os.makedirs(args.exp_dir)
if args.world_size == 4:
shutil.copy("./run_bear.sh", f"{args.exp_dir}/run.sh")
else:
shutil.copy("./run.sh", f"{args.exp_dir}/run.sh")
#if args.world_size == 4:
# shutil.copy("./run_bear.sh", f"{args.exp_dir}/run.sh")
#else:
# shutil.copy("./run.sh", f"{args.exp_dir}/run.sh")
args_dict = vars(args)
f = open(f"{args.exp_dir}/config.yaml", 'w')
for k, v in args_dict.items():