keras/src/callbacks/model_checkpoint.py/ModelCheckpoint/on_train_batch_end
class ModelCheckpoint:    def on_train_batch_end(self, batch, logs=None):
        if self._should_save_on_batch(batch):
            self._save_model(epoch=self._current_epoch, batch=batch, logs=logs)