Resume the optimizers and schedulers for training
(self, resume_state)
| 109 | torch.save(state, save_path) |
| 110 | |
| 111 | def resume_training(self, resume_state): |
| 112 | """Resume the optimizers and schedulers for training""" |
| 113 | resume_optimizers = resume_state['optimizers'] |
| 114 | resume_schedulers = resume_state['schedulers'] |
| 115 | assert len(resume_optimizers) == len(self.optimizers), 'Wrong lengths of optimizers' |
| 116 | assert len(resume_schedulers) == len(self.schedulers), 'Wrong lengths of schedulers' |
| 117 | for i, o in enumerate(resume_optimizers): |
| 118 | self.optimizers[i].load_state_dict(o) |
| 119 | for i, s in enumerate(resume_schedulers): |
| 120 | self.schedulers[i].load_state_dict(s) |