snntorch:在Python中使用尖峰神经网络进行深度学习 源码

介绍 snnTorch是一个Python软件包,用于使用尖峰神经网络执行基于梯度的学习。 它无需重新设计轮子,而是位于PyTorch的顶部,并利用其GPU加速的张量计算。 预先设计的尖峰神经元模型无缝集成在PyTorch框架内,可以视为循环激活单元。 snTorch结构 snnTorch包含以下组件: 成分 描述 像torch.nn这样尖峰的神经元库,与autograd紧密集成 SNN通常使用的反向传播变化 用于生成峰值和数据转换的库 使用matplotlib和赛璐ul的基于峰值的数据的可视化工具 可选的替代梯度函数 数据集实用程序功能 snnTorch设计为可直观地与PyTorc