netF = arch.VGGFeatureExtractor(feature_layer=feature_layer, use_bn=use_bn, \
use_input_norm=True, device=device)
if gpu_ids:
netF = nn.DataParallel(netF).cuda()
netF.eval() // No need to train
return netF
After Change
netF = arch.VGGFeatureExtractor(feature_layer=feature_layer, use_bn=use_bn, \
use_input_norm=True, device=device)
if gpu_ids:
netF = nn.DataParallel(netF)
netF.eval() // No need to train
return netF