我在使用 <<用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
个赞
请登录后评论
TOP
切换版块
看看自己的数据吧 维度问题
解决了吗?我也是相同的问题
如果是第一维大小不同,可能是batch_size大小不一导致。
输入数据的形状错误。