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):
def _calculate_metrics(self) -> MutableMapping[str, Any]:
logs = {}
if self._track_lr:
logs['learning_rate'] = self._calculate_lr()
# TODO(b/149030439): disable LR reporting.
# if self._track_lr:
# logs['learning_rate'] = self._calculate_lr()
return logs
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