megvii-research / MOTRv2

[CVPR2023] MOTRv2: Bootstrapping End-to-End Multi-Object Tracking by Pretrained Object Detectors
Other
359 stars 45 forks source link

用两个A100训练,只能达到65 Training with two A100s can only reach 65 #66

Open xun9167 opened 6 months ago

xun9167 commented 6 months ago

我在两个40G的A100上训练,最后的指标HOTA只有65,请问是因为训练时GPU数量不足的原因吗? I trained on two 40G A100s, and the final HOTA indicator was only 65. May I ask if this is due to the insufficient number of GPUs during training?

HELLORPG commented 5 months ago

原文采用的 batch size 应该是 8,如果使用两张 GPU 的话应该是无法得到同样的 batch size 的,这确实会对最终的结果产生影响。