def epoch_end(self, epoch):
    "End epoch.\n\n    Args:\n      epoch: `int`, the epoch number.\n\n    Raises:\n      ValueError: if we've not begun an epoch, or `epoch` number does not match.\n    "
    if (self._current_epoch != epoch):
        raise ValueError('epoch_end expected %s but got %s.', self._current_epoch, epoch)
    self._current_epoch = None
End epoch.

Args:
  epoch: `int`, the epoch number.

Raises:
  ValueError: if we've not begun an epoch, or `epoch` number does not match.
def epoch_end(self, epoch):
    "End epoch.\n\n    Args:\n      epoch: `int`, the epoch number.\n\n    Raises:\n      ValueError: if we've not begun an epoch, or `epoch` number does not match.\n    "
    if (self._current_epoch != epoch):
        raise ValueError('epoch_end expected %s but got %s.', self._current_epoch, epoch)
    self._current_epoch = None