随笔分类 -  SNN

同一个train.py，没改任何代码，一会能跑，一会不能跑
摘要：问题描述 跑一个模型，加载预训练权重，只进行推理，没有修改任何代码，刚开始直接运行可以得到推理结果，然后再跑，程序又卡住了；然后我使用调试，一会能跑，一会跑到第五六个batch时，又卡住了。 解决方法 修改dataloader的num_workers。我程序原来workers设为的8，后面改为4就可 阅读全文
posted @ 2023-12-03 14:18 Kurie 阅读(45) 评论(0) 推荐(0) 编辑
SNN知识储备-ANN2SNN
摘要：1. SpiCalib 1.1 针对的问题 在CNN转SNN时，通常会有较大的性能下降和时延 以前的工作主要集中在简单的分类任务上，忽略了SNN神经元的发电率应当是对ANN神经元激活值的精确近似 1.2 方法 提出了Spike Calibration来消除离散脉冲对输出分布的有害影响 使用burst 阅读全文
posted @ 2023-04-07 11:06 Kurie 阅读(183) 评论(0) 推荐(0) 编辑
修改训练数据集：网络原本使用MNIST数据集进行训练，现改为使用CIFAR10来进行训练。报错：RuntimeError: Given groups=1, weight of size [6, 1, 13, 13], expected input[1, 3, 32, 32] to have 1 channels, but got 3 channels instead
摘要：原因：通道数不匹配，MINIST使用灰度图像，CIFAR10使用RGB图像。 解决方式：加载图像时，做一下转换。 阅读全文
posted @ 2022-12-02 13:17 Kurie 阅读(2583) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示