Resolve #62 BERT training shows 0 loss (#75)

asyml · Jun 26, 2019 · 17c44ac · 17c44ac
1 parent 57ad85d
commit 17c44ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/bert/bert_classifier_main.py b/examples/bert/bert_classifier_main.py
@@ -177,7 +177,7 @@ def _train_epoch():
 
             dis_steps = config_data.display_steps
             if dis_steps > 0 and step % dis_steps == 0:
-                logging.info("step: %d; loss: %d", step, loss)
+                logging.info("step: %d; loss: %f", step, loss)
 
             eval_steps = config_data.eval_steps
             if eval_steps > 0 and step % eval_steps == 0: