Merge pull request #488 from helmutg/patch-1

FeatureExtractor: avoid loading weights twice
2022-02-23 20:57:59 +08:00 · 2022-02-23 20:57:59 +08:00 · 40bfa1ae77
parent 6987d99074 42185c4beb
commit 40bfa1ae77
1 changed files with 1 additions and 1 deletions
--- a/torchreid/utils/feature_extractor.py
+++ b/torchreid/utils/feature_extractor.py
@ -71,7 +71,7 @@ class FeatureExtractor(object):
        model = build_model(
            model_name,
            num_classes=1,
-            pretrained=True,
+            pretrained=not (model_path and check_isfile(model_path)),
            use_gpu=device.startswith('cuda')
        )
        model.eval()