if self.early_stopping.early_stop:
    print("Early stopping")
    break