用自己的数据集报错shape not match
收藏
我在使用 <<用Paddle实现高性能目标检测网络：ShuffleNetV2-YOLOv3>> 训练自己做的数据集时候报错
File "train.py", line 451, in train
load_pretrained_params(exe, train_program)
File "train.py", line 412, in load_pretrained_params
main_program=program)
RuntimeError: Shape not matching: the Program requires a parameter with a shape of ((21, 512, 1, 1)), while the loaded parameter (namely [ conv2d_61.w_0 ]) has a shape of ((75, 512, 1, 1)).
新手一枚，劳烦各位答疑解惑
0
收藏
请登录后评论
RuntimeError: Shape not matching
这个是shape不一致报错了。要求的shape是(21, 512, 1, 1)，但实际loaded的是(75, 512, 1, 1)。可以检查下自己的数据集。
如果没理解错的话四个维度分别表示n，h，w，c，所以你的批大小没对应上，试试应该把batch_size设置成21或者检查一下喂数据的时候是否出了问题