FeatureExtractor: avoid loading weights twice
When passing a model_path to the FeatureExtractor, weights are loaded twice. One the pretrained once and then the given ones. Avoid the unnecessary load.pull/488/head
parent
6987d99074
commit
42185c4beb
|
@ -71,7 +71,7 @@ class FeatureExtractor(object):
|
||||||
model = build_model(
|
model = build_model(
|
||||||
model_name,
|
model_name,
|
||||||
num_classes=1,
|
num_classes=1,
|
||||||
pretrained=True,
|
pretrained=not (model_path and check_isfile(model_path)),
|
||||||
use_gpu=device.startswith('cuda')
|
use_gpu=device.startswith('cuda')
|
||||||
)
|
)
|
||||||
model.eval()
|
model.eval()
|
||||||
|
|
Loading…
Reference in New Issue