(loggers, loaders, model, optimizer, scheduler)
| 47 | |
| 48 | |
| 49 | def train_example(loggers, loaders, model, optimizer, scheduler): |
| 50 | start_epoch = 0 |
| 51 | if cfg.train.auto_resume: |
| 52 | start_epoch = load_ckpt(model, optimizer, scheduler) |
| 53 | if start_epoch == cfg.optim.max_epoch: |
| 54 | logging.info('Checkpoint found, Task already done') |
| 55 | else: |
| 56 | logging.info('Start from epoch {}'.format(start_epoch)) |
| 57 | |
| 58 | num_splits = len(loggers) |
| 59 | for cur_epoch in range(start_epoch, cfg.optim.max_epoch): |
| 60 | train_epoch(loggers[0], loaders[0], model, optimizer, scheduler) |
| 61 | loggers[0].write_epoch(cur_epoch) |
| 62 | if is_eval_epoch(cur_epoch): |
| 63 | for i in range(1, num_splits): |
| 64 | eval_epoch(loggers[i], loaders[i], model) |
| 65 | loggers[i].write_epoch(cur_epoch) |
| 66 | if is_ckpt_epoch(cur_epoch): |
| 67 | save_ckpt(model, optimizer, scheduler, cur_epoch) |
| 68 | for logger in loggers: |
| 69 | logger.close() |
| 70 | if cfg.train.ckpt_clean: |
| 71 | clean_ckpt() |
| 72 | |
| 73 | logging.info('Task done, results saved in {}'.format(cfg.out_dir)) |
| 74 | |
| 75 | |
| 76 | register_train('example', train_example) |
nothing calls this directly
no test coverage detected