自己的训练集报错 shape not match
z807048861 发布于2020-04 浏览:2210 回复:4
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
个赞
共4条回复 最后由七年期限回复于2021-04
#5七年期限回复于2021-04

看看自己的数据吧  维度问题

0
#4回复于2021-04

解决了吗?我也是相同的问题

0
#3189******30回复于2020-04

如果是第一维大小不同,可能是batch_size大小不一导致。

0
#2189******30回复于2020-04

输入数据的形状错误。

0
TOP
切换版块