用自己的数据集报错shape not match
z807048861 发布于2020-04-13 22:51 浏览:383 回复:2
0
收藏

我在使用 <<用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
个赞
共2条回复 最后由鹿鼎记肯定回复于2020-04-20 14:25
#3鹿鼎记肯定回复于2020-04-20 14:25:20

如果没理解错的话四个维度分别表示n,h,w,c,所以你的批大小没对应上,试试应该把batch_size设置成21或者检查一下喂数据的时候是否出了问题

0
#2xiaohongyd回复于2020-04-17 14:53:04

RuntimeError: Shape not matching

这个是shape不一致报错了。要求的shape是(21, 512, 1, 1),但实际loaded的是(75, 512, 1, 1)。可以检查下自己的数据集。

0
TOP
切换版块