Commit e5bbd971 authored by Frisinghelli Daniel's avatar Frisinghelli Daniel
Browse files

Fix logging of loss.

parent 4b52908c
......@@ -1359,7 +1359,7 @@ class NetworkTrainer(BaseConfig):
isinstance(self.model, nn.DataParallel) else self.model.epoch)
overall = ('Mean accuracy: {:.2f}%'.format(np.mean(accuracy) * 100) if
self.classification else
'Mean loss: {:.2f}%'.format(np.mean(loss)))
'Mean loss: {:.2f}'.format(np.mean(loss)))', '.join(['Epoch: {:d}'.format(epoch), overall]))
if self.classification and return_pred:
Supports Markdown
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