Fix loss printing

TheCodez · May 29, 2019 · 46ff43f · 46ff43f
1 parent c817ec8
commit 46ff43f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train_kitti.py b/train_kitti.py
@@ -161,8 +161,8 @@ def log_validation_results(engine):
         loss = metrics['loss']
         iou = metrics['mIoU']
 
-        pbar.log_message('Validation results - Epoch: [{}/{}]: Loss: {:.4f}, mIoU: {:.1f}'
-                         .format(loss, engine.state.epoch, engine.state.max_epochs, iou * 100.0))
+        pbar.log_message('Validation results - Epoch: [{}/{}]: Loss: {:.2e}, mIoU: {:.1f}'
+                         .format(engine.state.epoch, engine.state.max_epochs, loss, iou * 100.0))
 
     @trainer.on(Events.EXCEPTION_RAISED)
     def handle_exception(engine, e):