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): def save_args(args):
if not os.path.exists(args.exp_dir): if not os.path.exists(args.exp_dir):
os.makedirs(args.exp_dir) os.makedirs(args.exp_dir)
if args.world_size == 4: #if args.world_size == 4:
shutil.copy("./run_bear.sh", f"{args.exp_dir}/run.sh") # shutil.copy("./run_bear.sh", f"{args.exp_dir}/run.sh")
else: #else:
shutil.copy("./run.sh", f"{args.exp_dir}/run.sh") # shutil.copy("./run.sh", f"{args.exp_dir}/run.sh")
args_dict = vars(args) args_dict = vars(args)
f = open(f"{args.exp_dir}/config.yaml", 'w') f = open(f"{args.exp_dir}/config.yaml", 'w')
for k, v in args_dict.items(): for k, v in args_dict.items():