Merge pull request #47006 from lgeiger:cleanup-backup-callback

PiperOrigin-RevId: 356787726
Change-Id: I75651ea2fa00ecc515b3e7550f7b1fc8c8d558ab

tensorflow/python/keras/callbacks.py

@@ -1628,9 +1628,6 @@ class BackupAndRestore(Callback):
     # restore checkpoint at on_train_begin().
     self._chief_worker_only = False
 
-  def set_model(self, model):
-    self.model = model
-
   def on_train_begin(self, logs=None):
     # TrainingState is used to manage the training state needed for
     # failure-recovery of a worker in training.