mmpose 我想把训练一个基于deeppose的mobilenetv3模型，然后训练的时候出错了，能帮我看看哪里出问题了吗？

我想把训练一个基于deeppose的mobilenetv3模型，然后训练的时候出错了，能帮我看看哪里出问题了吗？

Open zeckireck opened this issue 3 years ago • 4 comments

这个是报的错

这个是配置文件的截图

Aug 31 '22 08:08 zeckireck

We recommend using English or English & Chinese for issues so that we could have broader discussion.

Aug 31 '22 08:08 mm-assistant[bot]

请问有没有训练过我们官方提供的config文件？排除一下是环境配置的问题

Aug 31 '22 08:08 jin-s13

嗯，能正常训练基于top-down的mobilenetv3

Aug 31 '22 09:08 zeckireck

这个 bug 可能是因为 FC 层 input, weight 维度不一致导致的。 MobileNetV3 arch='small' 的时候，输出的 feature map 维度数是96，keypoint_heat.in_channels 可以改成96试一下。或者可以尝试 arch='big'，对应的 backbone 输出的 feature map 维度数是 160

Sep 01 '22 04:09 Ben-Louis

mmpose mmpose copied to clipboard

我想把训练一个基于deeppose的mobilenetv3模型，然后训练的时候出错了，能帮我看看哪里出问题了吗？

mmpose
mmpose copied to clipboard