Skip to content

Commit aff514a

Browse files
saberkunallenwang28
authored andcommitted
Disable LR reporting due to reporting bug.
PiperOrigin-RevId: 304315213
1 parent 72238e5 commit aff514a

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

official/vision/image_classification/callbacks.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -128,8 +128,9 @@ def on_epoch_end(self,
128128

129129
def _calculate_metrics(self) -> MutableMapping[str, Any]:
130130
logs = {}
131-
if self._track_lr:
132-
logs['learning_rate'] = self._calculate_lr()
131+
# TODO(b/149030439): disable LR reporting.
132+
# if self._track_lr:
133+
# logs['learning_rate'] = self._calculate_lr()
133134
return logs
134135

135136
def _calculate_lr(self) -> int:

0 commit comments

Comments
 (0)