分布式训练问题 - Githubissues

weiguangzhao / DF_MVR

From 2D Images to 3D Model: Weakly Supervised Multi-View Face Reconstruction with Deep Fusion

18 stars 1 forks source link

Closed nlnkjb closed 2 weeks ago

nlnkjb commented 2 weeks ago

您好，就是我启用分布式训练的时候，在epoch大于100以后会调用bisenet网络，但是这时候传入网络的渲染图像的维度是[2,16,3,224,224],可是bisenet网络里我看您写的代码在输入为5维的时候没法处理呀

nlnkjb commented 2 weeks ago

已解决，代码在100轮以后计算mask损失那一块有问题，取消分布式训练，注释掉render_img_face = render_img_face.unsqueeze(0).cuda()后代码正常训练，