(self, is_best, filename, best_model)
| 28 | |
| 29 | |
| 30 | def save(self, is_best, filename, best_model): |
| 31 | print('Saving checkpoint at "%s"' % filename) |
| 32 | torch.save(self, filename) |
| 33 | if is_best: |
| 34 | print('Saving the best model at "%s"' % best_model) |
| 35 | shutil.copyfile(filename, best_model) |
| 36 | print('\n') |
| 37 | |
| 38 | |
| 39 | def load(self, filename): |
nothing calls this directly
no outgoing calls
no test coverage detected