딥러닝

AlexNet vs SNN torch /w MNist dataset

Sumin Lim 2023. 9. 28. 19:12
반응형

mnist 를 alexNet으로 돌려본 적 있는데 colab기본 cpu에서도 엄청 오래걸렸었다. 거의 50분에서 한시간?

근데 SNN_torch로 돌려보니 엄청 빠르다.

정확도도 비슷하거나 더 높고 

내가 돌려본 AlexNet코드 mnist: 
https://github.com/imsumin018/daily-code-snippet/blob/main/AlexNet_Mnist_final_01.ipynb

내가 돌려본 SNNtorch 코드 :

https://github.com/imsumin018/daily-code-snippet/blob/main/StudySNN_withMNIST_tutorial_5_neuromorphic_datasets.ipynb

https://snntorch.readthedocs.io/en/latest/tutorials/index.html

 

 

Tutorials — snntorch 0.7.0 documentation

© Copyright 2021, Jason K. Eshraghian. Revision b756d17f.

snntorch.readthedocs.io

 

반응형