keras/src/callbacks/backup_and_restore.py/BackupAndRestore/on_train_batch_end
class BackupAndRestore:    def on_train_batch_end(self, batch, logs=None):
        if self._should_save_on_batch(batch):
            self._save_model()