I was writing my own callback to stop training based on some custom condition. EarlyStopping has this to stop the training once condition is met:
self.model.s