From 81d561e0e31614a3e910ef6ef988e1462172ec9b Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Wed, 13 Nov 2019 18:02:43 +0200
Subject: [PATCH] image_classifier.py: fix printout when using early-exit
When performing EE validation, the validation loop prints
an annoying and redundant log of the iteration number - remove this.
---
distiller/apputils/image_classifier.py | 1 -
1 file changed, 1 deletion(-)
diff --git a/distiller/apputils/image_classifier.py b/distiller/apputils/image_classifier.py
index ab3b5b9..907772c 100755
--- a/distiller/apputils/image_classifier.py
+++ b/distiller/apputils/image_classifier.py
@@ -660,7 +660,6 @@ def _validate(data_loader, model, criterion, loggers, args, epoch=-1):
('Top5', classerr.value(5))])
else:
stats_dict = OrderedDict()
- stats_dict['Test'] = validation_step
for exitnum in range(args.num_exits):
la_string = 'LossAvg' + str(exitnum)
stats_dict[la_string] = args.losses_exits[exitnum].mean
--
GitLab