Internal change
PiperOrigin-RevId: 357849502 Change-Id: I94253831b959ee71cb9f396c77158d7912cfa7d5
This commit is contained in:
parent
54de9ec7e6
commit
a26dc6a474
@ -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):
|
||||
|
Loading…
x
Reference in New Issue
Block a user