search

yingxin-jia / SuperGlue-pytorch

[SuperGlue: Learning Feature Matching with Graph Neural Networks] This repo includes PyTorch code for training the SuperGlue matching network on top of SIFT keypoints and descriptors.
Other
545 stars 126 forks source link

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

Open ahong007007 opened 4 years ago

ahong007007 commented 4 years ago

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

image

ZhouAo-ZA commented 4 years ago

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

Jazihars commented 2 years ago

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

ZhouAo-ZA commented 2 years ago

自己的仓库

thanks!