文章目录 背景 代码 报错 原因 解决 网络结构（修改前） 网络结构（修改后） 类似的错误 参考 背景 人脸40个属性预测，原网络为Arcface，最后一层输出512维特征，修改后的网络输出为40个属性的二分类特征输出。 代码 loss.append(criterion(output, target, )) 报错 RuntimeError: Dimension out of range (expected to be in range of [-1, 0]</
超级会员免费看
在尝试将Arcface网络用于人脸40个属性预测时，遇到RuntimeError，原因是网络输出与标签尺寸不匹配。解决方案是调整网络结构，增加一层以确保输出与目标shape一致，满足40个属性的二分类需求。
订阅专栏 解锁全文
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
