B导，我用分布式训练时将distributed和sync_bn都改为True时报错

bubbliiiing / yolox-pytorch

这是一个yolox-pytorch的源码，可以用于训练自己的模型。

Apache License 2.0

877 stars 184 forks source link

B导，我用分布式训练时将distributed和sync_bn都改为True时报错 #128

Open mingshan3 opened 1 year ago

mingshan3 commented 1 year ago

报错具体为：RuntimeError: expected device cuda:1 and dtype Float but got device cuda:0 and dtype Float 我的python是3.7 torch为1.2 JH AI5FVBD%0GXF1YW7L3 V QQ截图20230301163043