Fix issue in feature extraction when using a local checkpoint

pull/489/head
ryanking13 4 years ago
parent f1f332fee5
commit f043c02c5a

@ -55,6 +55,7 @@ def create_model(
raise RuntimeError('Unknown model (%s)' % model_name)
if checkpoint_path:
load_checkpoint(model, checkpoint_path)
features_only = kwargs.get("features_only", False)
load_checkpoint(model, checkpoint_path, strict=not features_only)
return model

Loading…
Cancel
Save