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

Multi-gpu training #18

Closed Yiiiike closed 3 years ago

Yiiiike commented 3 years ago

您好,我在训练的过程中遇到了显存不够的问题。 请问可以使用多块gpu运行ann-to-snn代码吗?如果可以的话,使用方法是什么呢? 谢谢您。

Yiiiike commented 3 years ago

非常抱歉,我已经找到了多gpu训练的方法。 打扰您了。

Berumotto1 commented 1 year ago

请问怎么使用多gpu运行ann-to-snn的代码?