Make output directories, setup logging, Tensorboard, snapshot code.
(args)
| 68 | |
| 69 | |
| 70 | def prep_experiment(args): |
| 71 | """ |
| 72 | Make output directories, setup logging, Tensorboard, snapshot code. |
| 73 | """ |
| 74 | args.ngpu = torch.cuda.device_count() |
| 75 | args.best_record = {'mean_iu': -1, 'epoch': 0} |
| 76 | |
| 77 | |
| 78 | def calculate_iou(hist_data): |