Skip to content

GitLab

Commit 17263f30 authored by Vighnesh Birodkar's avatar Vighnesh Birodkar Committed by TF Object Detection Team
Browse files

Log the loss value in summary exporter. 

PiperOrigin-RevId: 363700908
parent 99572269
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
research/object_detection/model_lib_v2.py
View file @ 17263f30
......@@ -686,6 +686,7 @@ def train_loop(
'steps_per_sec': np.mean(steps_per_sec_list),
'steps_per_sec_p50': np.median(steps_per_sec_list),
'steps_per_sec_max': max(steps_per_sec_list),
'last_batch_loss': loss
}
mixed_precision = 'bf16' if kwargs['use_bfloat16'] else 'fp32'
performance_summary_exporter(metrics, mixed_precision)
......
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