This commit is contained in:
wty-yy
2025-12-03 09:43:16 +08:00
parent ad907089a8
commit 9f7db42f9f
12 changed files with 179 additions and 60 deletions

View File

@@ -58,6 +58,8 @@ def parse_args():
for param in parameters:
parser.add_argument(param['name'], **{k: v for k, v in param.items() if k != 'name'})
args = parser.parse_args()
if args.experiment_name is None:
args.experiment_name = f"exp"
if args.experiment_name is not None:
args.experiment_name = f"{args.task_name}_{args.experiment_name}"
else:
args.experiment_name = args.task_name
return args