Closed seobbro closed 2 years ago
I downloaded your pretrained weight "sew50_checkpoint_319.pth", and found that your weight does not include the key 'fc'. Can you check if the shared pretrained weight includes the fc weight?
import torch
pt = torch.load('D:/pt/pt/sew50_checkpoint_319.pth')
print(pt['model']['fc.weight'].shape)
print(pt['model']['fc.bias'].shape)
torch.Size([1000, 2048])
torch.Size([1000])
Thanks!
https://figshare.com/articles/software/Spike-Element-Wise-ResNet/14752998