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

pytorch 1.3 is not consistent with spikingjelly (need pytorch>=1.5) #26

Open xin-xin-xin opened 3 years ago

xin-xin-xin commented 3 years ago

Hi~

ValueError: ('cannot find tensor Size', torch.Size([16, 16, 320, 320]))

This error occours, I find the solution is : using pytorch 1.3, I tried this method, but the spikingjelly is dependent on pytorch>=1.5, so I don't know the execute solution~

Best wish Xin

cwq159 commented 3 years ago

You can use different environment. PyTorch>=1.5 will be supported in the future.