Преглед на файлове

minor change in classifier loading

Dimitri Korsch преди 4 години
родител
ревизия
1b0ad79db1
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      cvmodelz/classifiers/base.py

+ 1 - 1
cvmodelz/classifiers/base.py

@@ -63,7 +63,7 @@ class Classifier(chainer.Chain):
 
 		try:
 			# Case (2)
-			self.load_classifier(weights_file)
+			return self.load_classifier(weights_file)
 
 		except KeyError as e:
 			pass