| 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) |
| 88 | load_path = os.path.join(self._save_dir, load_filename) |
| 89 | assert os.path.exists( |
| 90 | load_path), 'Weights file not found. Have you trained a model!? We are not providing one' % load_path |
| 91 | |
| 92 | optimizer.load_state_dict(torch.load(load_path)) |
| 93 | print 'loaded optimizer: %s' % load_path |
| 94 | |
| 95 | def _save_network(self, network, network_label, epoch_label): |
| 96 | save_filename = 'net_epoch_%s_id_%s.pth' % (epoch_label, network_label) |