[BERT/TF2] Add convergence log

Author: meatybobby

TensorFlow2/LanguageModeling/BERT/official/modeling/model_training_utils.py

@@ -607,5 +607,6 @@ def _run_callbacks_on_batch_end(batch):
 
       if dllogging and perf_wo_n != 0:
         dllogging.logger.log(step=(), data={"throughput_train": perf_wo/perf_wo_n}, verbosity=Verbosity.DEFAULT)
+        dllogging.logger.log(step=(), data={"total_loss": training_summary['train_loss']}, verbosity=Verbosity.DEFAULT)
 
     return model