Skip to content

GitLab

Commit ea3e2f7b authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Disable LR reporting due to reporting bug. 

PiperOrigin-RevId: 304315213
parent 6d16ae2e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
official/vision/image_classification/callbacks.py
View file @ ea3e2f7b
...@@ -128,8 +128,9 @@ class CustomTensorBoard(tf.keras.callbacks.TensorBoard): ...@@ -128,8 +128,9 @@ class CustomTensorBoard(tf.keras.callbacks.TensorBoard):
def _calculate_metrics(self) -> MutableMapping[str, Any]: def _calculate_metrics(self) -> MutableMapping[str, Any]:
logs = {} logs = {}
if self._track_lr: # TODO(b/149030439): disable LR reporting.
logs['learning_rate'] = self._calculate_lr() # if self._track_lr:
# logs['learning_rate'] = self._calculate_lr()
return logs return logs
def _calculate_lr(self) -> int: def _calculate_lr(self) -> int:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment