使用官方提供的模型作为预训练模型训练自己数据集报错
Created by: yeyupiaoling
我使用如下的官方中文训练模型作为预训练模型训练自己的数据集,报错,
错误信息如下:
Traceback (most recent call last):
File "train.py", line 118, in <module>
main()
File "train.py", line 114, in main
train()
File "train.py", line 109, in train
test_off=args.test_off)
File "/DeepSpeech/model_utils/model.py", line 307, in train
pre_epoch = self.init_from_pretrained_model(exe, train_program)
File "/DeepSpeech/model_utils/model.py", line 161, in init_from_pretrained_model
filename="params.pdparams")
File "/usr/local/lib/python2.7/dist-packages/paddle/fluid/io.py", line 784, in load_params
filename=filename)
File "/usr/local/lib/python2.7/dist-packages/paddle/fluid/io.py", line 668, in load_vars
filename=filename)
File "/usr/local/lib/python2.7/dist-packages/paddle/fluid/io.py", line 727, in load_vars
format(orig_shape, each_var.name, new_shape))
RuntimeError: Shape not matching: the Program requires a parameter with a shape of ((1312L, 3072L)), while the loaded parameter (namely [ layer_2_forward_fc_weight ]) has a shape of ((1312, 6144)).
Failed in training!