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
207 stars 58 forks source link

Cannot use multi card to test #24

Open Perfect-August opened 3 years ago

Perfect-August commented 3 years ago

您好,我的显卡是12G的显存,当我使用多卡运行ann-snn的时候,出现了以下错误:TypeError: zip argument #1 must support iteration,请问该如何解决? 微信截图_20210819155520 另外,请问您考虑添加IBT神经元吗?

cwq159 commented 3 years ago

抱歉,目前还不支持多卡,您可以尝试添加。 IBT还不支持,目前支持仅使用正脉冲的LIF(考虑到实际部署时芯片通常仅支持LIF和正脉冲),而IBT需要用到负脉冲,您也可以尝试添加。