SuperGlue-pytorch 请教，如何实现superglue多卡的训练？

请教，如何实现superglue多卡的训练？

Open ahong007007 opened this issue 4 years ago • 3 comments

在设置torch.nn.DataParallel(superglue)之后，出现数据合并错误。

Aug 24 '20 08:08 ahong007007

我也遇到了同样问题，还有就是多batch训练，请问你解决了吗 @ahong007007

Oct 07 '20 08:10 ZhouAo-ZA

我在自己的仓库里实现了一个SuperGlue单机多卡训练的脚本，可以参考@ahong007007 @ZhouAo-ZA

Dec 09 '21 09:12 Jazihars

自己的仓库

thanks!

Dec 10 '21 02:12 ZhouAo-ZA