experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

Browse Source 
PiperOrigin-RevId: 356787726
Change-Id: I75651ea2fa00ecc515b3e7550f7b1fc8c8d558ab
This commit is contained in:
TensorFlower Gardener 2021-02-10 11:54:00 -08:00
commit 4499fa94bc
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/keras/callbacks.py
View File

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