()
| 53 | |
| 54 | @logger.catch #在子线程或主线程中捕获异常 |
| 55 | def main(): |
| 56 | args = get_parser() |
| 57 | args.manual_seed = init_random_seed(args.manual_seed) |
| 58 | set_random_seed(args.manual_seed, deterministic=True) |
| 59 | |
| 60 | args.ngpus_per_node = torch.cuda.device_count() |
| 61 | args.world_size = args.ngpus_per_node * args.world_size |
| 62 | mp.spawn(main_worker, nprocs=args.ngpus_per_node, args=(args, )) |
| 63 | |
| 64 | |
| 65 | def main_worker(gpu, args): |
no test coverage detected