search

cwq159 / PyTorch-Spiking-YOLOv3

A PyTorch implementation of Spiking-YOLOv3. Two branches are provided, based on two common PyTorch implementation of YOLOv3(ultralytics/yolov3 & eriklindernoren/PyTorch-YOLOv3), with support for Spiking-YOLOv3-Tiny at present.
GNU General Public License v3.0
209 stars 58 forks source link

关于ann to snn转换后测试代码段的问题 #42

Open powerqjy opened 2 years ago

powerqjy commented 2 years ago

您好！ 请问ann_to_snn.py代码中最后的test段落， if opt.task == 'test': # (default) test normally snn_results, maps, firing_ratios = snn_evaluate(opt, data, ann, snn, opt.timesteps, batch_size, imgsz_test, opt.conf_thres, opt.iou_thres, opt.save_json, opt.single_cls, opt.augment) 函数snn_evaluate()测试过程中有两次内存爆炸的现象，请问这是为什么呢？ 我遍历snn_evaluate()代码没有发现有优化网络结构相关的部分，这部分却产生了巨量数据。注释掉以后发现timesteps可以无限提高。请问此函数除了测试模型准确率，还具备其他什么作用吗？是否可以省略此部分呢？

layne22 commented 2 years ago

请问有解决这个问题吗？

powerqjy commented 2 years ago

请问有解决这个问题吗？

我转出来发现只要测试就需要巨量内存，没得办法，可能论文里对神经元相关处理没说全。我放弃这个模型了。

cwq159 commented 2 years ago

这部分需要把输入图片复制timesteps份，因此有可能导致内存溢出，可以通过调低timesteps来减少复制数据。

powerqjy commented 2 years ago

这个程序无论是检测还是转换，内存是不断递增的，这个是为什么呢

---原始邮件--- 发件人: "Weiqian @.> 发送时间: 2022年7月26日(周二) 下午4:45 收件人: @.>; 抄送: @.**@.>; 主题: Re: [cwq159/PyTorch-Spiking-YOLOv3] 关于ann to snn转换后测试代码段的问题 (Issue #42)

这部分需要把输入图片复制timesteps份，因此有可能导致内存溢出，可以通过调低timesteps来减少复制数据。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

123456ZF commented 1 year ago

@powerqjy 你好，请问，在运行ann-snn.py文件时，出现了这个问题：raise ValueError(cannot find tensor Size, tensor.size())。我安装了pytorch1.6,python3.7,然后出现：warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))，这是怎么回事，是30系显卡的原因吗？请问您当初是怎么跑成功的吗？