Commit 46d3230b authored by Julian Treiber's avatar Julian Treiber

added allow_missing and ignore_extra for pretrained weights

parent c3dc2a23
......@@ -74,7 +74,7 @@ class ${tc.fileNameWithoutEnding}:
lastEpoch = epoch
param_file = file
logging.info("Loading pretrained weights: " + self._weights_dir_ + param_file)
network.load_parameters(self._weights_dir_ + param_file)
network.load_parameters(self._weights_dir_ + param_file, allow_missing=True, ignore_extra=True)
else:
logging.info("No pretrained weights available at: " + self._weights_dir_ + param_file)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment