def save_checkpoint(self):
"""Save checkpoint."""
self.ckpt.steps.assign(self.steps)
self.ckpt.epochs.assign(self.epochs)
self.ckp_manager.save(checkpoint_number=self.steps)
utils.save_weights(self._model,
self.saved_path + "model-{}.h5".format(self.steps))
def save_checkpoint(self):
"""Save checkpoint."""
self.ckpt.steps.assign(self.steps)
self.ckpt.epochs.assign(self.epochs)
self.ckp_manager.save(checkpoint_number=self.steps)
utils.save_weights(
self._generator,
self.saved_path + "generator-{}.h5".format(self.steps))
utils.save_weights(
self._discriminator,
self.saved_path + "discriminator-{}.h5".format(self.steps))