diff --git a/distiller/apputils/image_classifier.py b/distiller/apputils/image_classifier.py index ab3b5b9dcd7136ae31c7a4f8e76c7d5e021a7986..907772cc5b21a39b4fc1bf725ffe5e96e180e570 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