search

yun-liu / RCF-PyTorch

Richer Convolutional Features for Edge Detection
Other
69 stars 10 forks source link

State_dict missing keys problem #7

Open fdy61 opened 1 year ago

fdy61 commented 1 year ago

I use the vgg16convs.mat the link provided to train 10 rounds.And I use the epoch10.pth to test,it will report an error ,saying missing keys,But I had checked that no keys had been missed.

6b3382b68457dd729a5b55256ef31a0
yun-liu commented 1 year ago 
checkpoint = torch.load('/path/to/your/checkpoint')
model.load_state_dict(checkpoint['state_dict'])
fdy61 commented 1 year ago

It worked.Thank you so much.