def on_epoch_end(self, epoch):
    super(CNNFinalTrainer, self).on_epoch_end(epoch)
    self.model.on_epoch_end(epoch)
    self.objective.on_epoch_end(epoch)
def on_epoch_end(self, epoch):
    super(CNNFinalTrainer, self).on_epoch_end(epoch)
    self.model.on_epoch_end(epoch)
    self.objective.on_epoch_end(epoch)