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

fixed loading for training from scratch

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

+ 3 - 0
cvmodelz/classifiers/base.py

@@ -77,6 +77,9 @@ class Classifier(chainer.Chain):
 
 	def load_classifier(self, weights_file: str):
 
+		if weights_file is None:
+			return
+
 		if isinstance(weights_file, io.BufferedIOBase):
 			assert not weights_file.closed, "The weights file was already closed!"
 			weights_file.seek(0)