(self, epoch, iteration)
| 95 | self.state.register(**kwargs) |
| 96 | |
| 97 | def update_iteration(self, epoch, iteration): |
| 98 | self.state.epoch = epoch |
| 99 | self.state.iteration = iteration |
| 100 | |
| 101 | def save_checkpoint(self, path): |
| 102 | logger.info("Saving checkpoint to file {}".format(path)) |