多GPU训练问题 - Githubissues

CaptainEven / MCMOT

Real time one-stage multi-class & multi-object tracking based on anchor-free detection and ReID

MIT License

390 stars 83 forks source link

Open ZacharyZgw opened 4 years ago

ZacharyZgw commented 4 years ago

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

CaptainEven commented 4 years ago

@ZacharyZgw 用的哪个版本的PyTorch?

ZacharyZgw commented 4 years ago

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