search

fangwei123456 / Spike-Element-Wise-ResNet

Deep Residual Learning in Spiking Neural Networks
Mozilla Public License 2.0
140 stars 21 forks source link

Can you provide ImageNet-pretrained weight of the fc-layer? #8

Closed seobbro closed 2 years ago

fangwei123456 commented 2 years ago

https://figshare.com/articles/software/Spike-Element-Wise-ResNet/14752998

seobbro commented 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?

fangwei123456 commented 2 years ago 
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])
seobbro commented 2 years ago

Thanks!