From e46b2a0a23c5f479bd810af7ff2d4a6f08f0056d Mon Sep 17 00:00:00 2001
From: KunHuang <hunterkun94@outlook.com>
Date: Mon, 9 Sep 2019 23:04:51 +0800
Subject: [PATCH] Bugfix - Error always raised in training loop after #375
(#377)
---
distiller/apputils/image_classifier.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/distiller/apputils/image_classifier.py b/distiller/apputils/image_classifier.py
index c5cd3f6..001afd1 100755
--- a/distiller/apputils/image_classifier.py
+++ b/distiller/apputils/image_classifier.py
@@ -156,7 +156,7 @@ class ClassifierCompressor(object):
msglogger.error(
'epoch count is too low, starting epoch is {} but total epochs set to {}'.format(
self.start_epoch, self.ending_epoch))
- raise ValueError('Epochs parameter is too low. Nothing to do.')
+ raise ValueError('Epochs parameter is too low. Nothing to do.')
# Load the datasets lazily
self.load_datasets()
--
GitLab