Internal change

PiperOrigin-RevId: 357849502
Change-Id: I94253831b959ee71cb9f396c77158d7912cfa7d5
This commit is contained in:
A. Unique TensorFlower 2021-02-16 18:29:24 -08:00 committed by TensorFlower Gardener
parent 54de9ec7e6
commit a26dc6a474

View File

@ -2428,8 +2428,6 @@ class TensorBoard(Callback, version_utils.TensorBoardVersionSelector):
lr_schedule = getattr(self.model.optimizer, 'lr', None)
if isinstance(lr_schedule, learning_rate_schedule.LearningRateSchedule):
logs['learning_rate'] = lr_schedule(self.model.optimizer.iterations)
elif lr_schedule is not None:
logs['learning_rate'] = lr_schedule
return logs
def _compute_steps_per_second(self):