(path)
| 1101 | print(f"{metric.upper():<8}: {value1:.4f} -> {value2:.4f} ({diff_str})") |
| 1102 | |
| 1103 | def create_dir(path): |
| 1104 | if os.path.exists(path): |
| 1105 | shutil.rmtree(path) |
| 1106 | os.makedirs(path) |
| 1107 | |
| 1108 | def main(local_rank: int, world_rank, world_size: int, cfg: Config): |
| 1109 | runner = Runner(local_rank, world_rank, world_size, cfg) |