zhouchenlin2096 / Spikingformer

Spikingformer: Spike-driven Residual Learning for Transformer-based Spiking Neural Network
Apache License 2.0
76 stars 5 forks source link

TypeError: __init__() got an unexpected keyword argument 'pretrained_cfg_overlay' #2

Closed LockFaiz closed 11 months ago

LockFaiz commented 1 year ago

Unexpected 'pretrained_cfg_overlay' error occurs

image

when I test model.py with timm 0.9.3.dev0, This error seems fixed if I add "pretrained_cfgoverlay=None" in the "\_init__" of class "vit_snn" in model.py

zhouchenlin2096 commented 1 year ago

Try ”timm==0.6.12“