|
@@ -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)
|