search

fangwei123456 / spikingjelly

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.
https://spikingjelly.readthedocs.io
Other
1.4k stars 244 forks source link

SNN中突触权重的获取和加载方法 #582

Open YuqingXiong opened 2 months ago

YuqingXiong commented 2 months ago

在ANN转换SNN的教程文档中：https://spikingjelly.readthedocs.io/zh-cn/latest/activation_based/ann2snn.html#id8

链接中提出需要对参数进行BN层的融合和缩放，这部分操作似乎在 Converter 类中的 fuse和set_voltagehook 函数中，那么最后得到的突触参数值要如何获得呢？另一方面，如果我想加载自己的SNN模型参数（注意不是ANN参数）又该如何做呢？

YuqingXiong commented 2 months ago

我在以前的 closed issue 中发现了类似的问题：https://github.com/fangwei123456/spikingjelly/issues/178

Met4physics commented 1 month ago

ANN-SNN的工作建议自己实现，会比spikingjelly目前的方便很多