(args)
| 153 | |
| 154 | |
| 155 | def init_summarywriter(args): |
| 156 | writer = None |
| 157 | if int(os.environ.get('RANK', 0)) == 0: |
| 158 | os.makedirs(args.model_dir, exist_ok=True) |
| 159 | writer = SummaryWriter(args.tensorboard_dir) |
| 160 | return writer |
| 161 | |
| 162 | |
| 163 | def save_model(model, model_name, info_dict): |