@TOCpytorch:
Traceback (most recent call last):
File “F:/weakalign-master/demo.py”, line 71, in
model.FeatureExtraction.state_dict()[name].copy_(checkpoint[‘state_dict’][‘FeatureExtraction.’ + name])
KeyError: ‘FeatureExtraction.model.1.num_batches_tracked’
if args.model != ‘’:
checkpoint = torch.load(args.model, map_location=lambda storage, loc: storage)
checkpoint[‘state_dict’] = OrderedDict([(k.replace(‘vgg’, ‘model’), v) for k, v in checkpoint[‘state_dict’].items()])
for name, param in model.FeatureExtraction.state_dict().items():
model.FeatureExtraction.state_dict()[name].copy_(checkpoint['state_dict']['FeatureExtraction.' + name])
for name, param in model.FeatureRegression.state_dict().items():
model.FeatureRegression.state_dict()[name].copy_(checkpoint['state_dict']['FeatureRegression.' + name])
for name, param in model.FeatureRegression2.state_dict().items():
model.FeatureRegression2.state_dict()[name].copy_(checkpoint['state_dict']['FeatureRegression2.' + name])
求助!!!