(args: APNamespace)
| 518 | |
| 519 | |
| 520 | def main(args: APNamespace): |
| 521 | print("Argument Parser Options") |
| 522 | print("-" * 45) |
| 523 | for arg in vars(args): |
| 524 | attr = getattr(args, arg) |
| 525 | attr = attr if attr is not None else "None" |
| 526 | print(f" {arg:<20}: {attr:<40}") |
| 527 | print("-" * 45) |
| 528 | args.config_path, args.output_path, args.data_path, args.checkpoint_path, args.resume = setup_dirs(args) |
| 529 | if args.dist: |
| 530 | init_process_group(backend='nccl') |
| 531 | main_worker(args) |
| 532 | destroy_process_group() |
| 533 | else: |
| 534 | main_worker(args) |
| 535 | |
| 536 | |
| 537 | def main_worker(args: APNamespace): |
no test coverage detected