(self, optimizer, optimizer_label, epoch_label)
| 79 | assert False, "load not implemented" |
| 80 | |
| 81 | def _save_optimizer(self, optimizer, optimizer_label, epoch_label): |
| 82 | save_filename = 'opt_epoch_%s_id_%s.pth' % (epoch_label, optimizer_label) |
| 83 | save_path = os.path.join(self._save_dir, save_filename) |
| 84 | torch.save(optimizer.state_dict(), save_path) |
| 85 | |
| 86 | def _load_optimizer(self, optimizer, optimizer_label, epoch_label): |
| 87 | load_filename = 'opt_epoch_%s_id_%s.pth' % (epoch_label, optimizer_label) |