what device do you used in ImageNet？

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

Closed 98ming closed 9 months ago

98ming commented 9 months ago

may I ask how long did you guys take to train thie model on ImageNet?

jkhu29 commented 9 months ago

20min/ep when dim==512 & layer==8

98ming commented 9 months ago

Thank you for your reply！ How many GPU did you use？And what is the name of the GPU？

jkhu29 commented 9 months ago

Thank you for your reply！ How many GPU did you use？And what is the name of the GPU？

8*A800

98ming commented 9 months ago

Thanks，it's so cool!