diff --git a/distiller/apputils/image_classifier.py b/distiller/apputils/image_classifier.py index abc1567a7a7211c67b5a87d95053747cd198be8e..47287fc9e6848a9ecba594dc339213f4389f8b76 100755 --- a/distiller/apputils/image_classifier.py +++ b/distiller/apputils/image_classifier.py @@ -732,8 +732,8 @@ def update_training_scores_history(perf_scores_history, model, top1, top5, epoch # Sort by sparsity as main sort key, then sort by top1, top5 and epoch perf_scores_history.sort(key=operator.attrgetter('params_nnz_cnt', 'top1', 'top5', 'epoch'), reverse=True) for score in perf_scores_history[:num_best_scores]: - msglogger.info('==> Best [Top1: %.3f Top5: %.3f Sparsity:%.2f Params: %d on epoch: %d]', - score.top1, score.top5, score.sparsity, score.params_nnz_cnt, score.epoch) + msglogger.info('==> Best [Top1: %.3f Top5: %.3f Sparsity:%.2f NNZ-Params: %d on epoch: %d]', + score.top1, score.top5, score.sparsity, -score.params_nnz_cnt, score.epoch) def earlyexit_loss(output, target, criterion, args):