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

关于使用多GPU运行ann_to_snn.py #41

Open ryikl opened 2 years ago

ryikl commented 2 years ago

你好，由于显存不够，我考虑使用多GPU运行ann_to_snn.py，一开始在device中写上要用的卡，但后边发现只用了一块卡。后来通过使用nn.DataParallel之后，会出现内存分布问题。我在一个issues中看到你聊到使用多GPU的问题，想请教一下你是怎么使用多GPU的，具体在哪里做改动。