search

BICLab / Spike-Driven-Transformer

Offical implementation of "Spike-driven Transformer" (NeurIPS2023)
https://openreview.net/forum?id=9FmolyOHi5
Apache License 2.0
212 stars 15 forks source link

pytorch version #5

Closed yang-yk closed 11 months ago

yang-yk commented 11 months ago

Thanks for your excellent work. I tried Pytorch 1.10.0 and 1.12.0 when I ran the code. However, it shows that the version of pytorch is wrong. Can you tell me the specific Pytorch version? Many Thanks.

jkhu29 commented 11 months ago

Please let me know your specific error message. But first, make sure that the local-rank argparse is accurate.

BICLab commented 11 months ago

Close this issue as there are no further questions, feel free to reopen it if you have more questions.