def on_evaluate_end(self):
    if self.visualize_when_val:
        self.visualize_pred()
    self.save_checkpoint()
def on_evaluate_end(self):
    if self.visualize_when_val:
        self.visualize_pred()
    self.save_checkpoint()