(args, accelerator, slurm_job_id)
| 48 | |
| 49 | |
| 50 | def update_note(args, accelerator, slurm_job_id): |
| 51 | args.note = ( |
| 52 | "v5" |
| 53 | + str(args.note) |
| 54 | + f"_{args.data.name}" |
| 55 | + f"_{args.model.name}" |
| 56 | + f"_bs{args.data.batch_size}" |
| 57 | + f"_wd{args.optim.wd}" |
| 58 | + f"_{accelerator.state.num_processes}g" |
| 59 | + f"_{slurm_job_id}" |
| 60 | ) |
| 61 | |
| 62 | return args.note |
| 63 | |
| 64 | |
| 65 | def has_text(args): |