MCMOT icon indicating copy to clipboard operation
MCMOT copied to clipboard
verified publisher icon CaptainEven

多GPU训练问题

Open ZacharyZgw opened this issue 5 years ago • 2 comments

你好！我在Visdrone-MOT数据集上进行训练，目标包含10个类别，采用多GPU训练出现以下问题(losses.py426行)，请问有什么解决方案吗？谢谢！单GPU训练是OK的。 image

ZacharyZgw avatar Jul 28 '20 12:07 ZacharyZgw

@ZacharyZgw 用的哪个版本的PyTorch?

CaptainEven avatar Jul 29 '20 01:07 CaptainEven

@CaptainEven 环境和FairMOT一样，PyTorch1.2.0。

ZacharyZgw avatar Jul 29 '20 01:07 ZacharyZgw