1079863482 / paddle2torch_PPOCRv3

90 stars 8 forks source link

模型转换失败 #2

Open ChaoII opened 2 years ago

ChaoII commented 2 years ago

小哥,你好: 我尝试git下你的源码,转换PPOCRv3官方提供的模型,成功了!但是使用ppocr训练自己的识别模型却转换不成功,报错信息是,缺少很多:如下:

Missing key(s) in state_dict: "backbone.conv1.conv.weight", "backbone.conv1.bn.weight", "backbone.conv1.bn.bias", "backbone.conv1.bn.running_mean", "backbone.conv1.bn.running_var", "backbone.block_list.0.dw_conv.conv.weight", "backbone.block_list.0.dw_conv.bn.weight", "backbone.block_list.0.dw_conv.bn.bias", "backbone.block_list.0.dw_conv.bn.running_mean", "backbone.block_list.0.dw_conv.bn.running_var", "backbone.block_list.0.pw_conv.conv.weight", "backbone.block_list.0.pw_conv.bn.weight", "backbone.block_list.0.pw_conv.bn.bias", "backbone.block_list.0.pw_conv.bn.running_mean", "backbone.block_list.0.pw_conv.bn.running_var", "backbone.block_list.1.dw_conv.conv.weight", "backbone.block_list.1.dw_conv.bn.weight", "backbone.block_list.1.dw_conv.bn.bias", "backbone.block_list.1.dw_conv.bn.running_mean", "backbone.block_list.1.dw_conv.bn.running_var", "backbone.block_list.1.pw_conv.conv.weight", "backbone.block_list.1.pw_conv.bn.weight", "backbone.block_list.1.pw_conv.bn.bias", "backbone.block_list.1.pw_conv.bn.running_mean", "backbone.block_list.1.pw_conv.bn.running_var", "backbone.block_list.2.dw_conv.conv.weight", "backbone.block_list.2.dw_conv.bn.weight", "backbone.block_list.2.dw_conv.bn.bias", "backbone.block_list.2.dw_conv.bn.running_mean", "backbone.block_list.2.dw_conv.bn.running_var", "backbone.block_list.2.pw_conv.conv.weight", "backbone.block_list.2.pw_conv.bn.weight", "backbone.block_list.2.pw_conv.bn.bias", "backbone.block_list.2.pw_conv.bn.running_mean", "backbone.block_list.2.pw_conv.bn.running_var", "backbone.block_list.3.dw_conv.conv.weight", "backbone.block_list.3.dw_conv.bn.weight", "backbone.block_list.3.dw_conv.bn.bias", "backbone.block_list.3.dw_conv.bn.running_mean", "backbone.block_list.3.dw_conv.bn.running_var", "backbone.block_list.3.pw_conv.conv.weight", "backbone.block_list.3.pw_conv.bn.weight", "backbone.block_list.3.pw_conv.bn.bias", "backbone.block_list.3.pw_conv.bn.running_mean", "backbone.block_list.3.pw_conv.bn.running_var", "backbone.block_list.4.dw_conv.conv.weight", "backbone.block_list.4.dw_conv.bn.weight", "backbone.block_list.4.dw_conv.bn.bias", "backbone.block_list.4.dw_conv.bn.running_mean", "backbone.block_list.4.dw_conv.bn.running_var", "backbone.block_list.4.pw_conv.conv.weight", "backbone.block_list.4.pw_conv.bn.weight", "backbone.block_list.4.pw_conv.bn.bias", "backbone.block_list.4.pw_conv.bn.running_mean", "backbone.block_list.4.pw_conv.bn.running_var", "backbone.block_list.5.dw_conv.conv.weight", "backbone.block_list.5.dw_conv.bn.weight", "backbone.block_list.5.dw_conv.bn.bias", "backbone.block_list.5.dw_conv.bn.running_mean", "backbone.block_list.5.dw_conv.bn.running_var", "backbone.block_list.5.pw_conv.conv.weight", "backbone.block_list.5.pw_conv.bn.weight", "backbone.block_list.5.pw_conv.bn.bias", "backbone.block_list.5.pw_conv.bn.running_mean", "backbone.block_list.5.pw_conv.bn.running_var", "backbone.block_list.6.dw_conv.conv.weight", "backbone.block_list.6.dw_conv.bn.weight", "backbone.block_list.6.dw_conv.bn.bias", "backbone.block_list.6.dw_conv.bn.running_mean", "backbone.block_list.6.dw_conv.bn.running_var", "backbone.block_list.6.pw_conv.conv.weight", "backbone.block_list.6.pw_conv.bn.weight", "backbone.block_list.6.pw_conv.bn.bias", "backbone.block_list.6.pw_conv.bn.running_mean", "backbone.block_list.6.pw_conv.bn.running_var", "backbone.block_list.7.dw_conv.conv.weight", "backbone.block_list.7.dw_conv.bn.weight", "backbone.block_list.7.dw_conv.bn.bias", "backbone.block_list.7.dw_conv.bn.running_mean", "backbone.block_list.7.dw_conv.bn.running_var", "backbone.block_list.7.pw_conv.conv.weight", "backbone.block_list.7.pw_conv.bn.weight", "backbone.block_list.7.pw_conv.bn.bias", "backbone.block_list.7.pw_conv.bn.running_mean", "backbone.block_list.7.pw_conv.bn.running_var", "backbone.block_list.8.dw_conv.conv.weight", "backbone.block_list.8.dw_conv.bn.weight", "backbone.block_list.8.dw_conv.bn.bias", "backbone.block_list.8.dw_conv.bn.running_mean", "backbone.block_list.8.dw_conv.bn.running_var", "backbone.block_list.8.pw_conv.conv.weight", "backbone.block_list.8.pw_conv.bn.weight", "backbone.block_list.8.pw_conv.bn.bias", "backbone.block_list.8.pw_conv.bn.running_mean", "backbone.block_list.8.pw_conv.bn.running_var", "backbone.block_list.9.dw_conv.conv.weight", "backbone.block_list.9.dw_conv.bn.weight", "backbone.block_list.9.dw_conv.bn.bias", "backbone.block_list.9.dw_conv.bn.running_mean", "backbone.block_list.9.dw_conv.bn.running_var", "backbone.block_list.9.pw_conv.conv.weight", "backbone.block_list.9.pw_conv.bn.weight", "backbone.block_list.9.pw_conv.bn.bias", "backbone.block_list.9.pw_conv.bn.running_mean", "backbone.block_list.9.pw_conv.bn.running_var", "backbone.block_list.10.dw_conv.conv.weight", "backbone.block_list.10.dw_conv.bn.weight", "backbone.block_list.10.dw_conv.bn.bias", "backbone.block_list.10.dw_conv.bn.running_mean", "backbone.block_list.10.dw_conv.bn.running_var", "backbone.block_list.10.pw_conv.conv.weight", "backbone.block_list.10.pw_conv.bn.weight", "backbone.block_list.10.pw_conv.bn.bias", "backbone.block_list.10.pw_conv.bn.running_mean", "backbone.block_list.10.pw_conv.bn.running_var", "backbone.block_list.11.dw_conv.conv.weight", "backbone.block_list.11.dw_conv.bn.weight", "backbone.block_list.11.dw_conv.bn.bias", "backbone.block_list.11.dw_conv.bn.running_mean", "backbone.block_list.11.dw_conv.bn.running_var", "backbone.block_list.11.se.conv1.weight", "backbone.block_list.11.se.conv1.bias", "backbone.block_list.11.se.conv2.weight", "backbone.block_list.11.se.conv2.bias", "backbone.block_list.11.pw_conv.conv.weight", "backbone.block_list.11.pw_conv.bn.weight", "backbone.block_list.11.pw_conv.bn.bias", "backbone.block_list.11.pw_conv.bn.running_mean", "backbone.block_list.11.pw_conv.bn.running_var", "backbone.block_list.12.dw_conv.conv.weight", "backbone.block_list.12.dw_conv.bn.weight", "backbone.block_list.12.dw_conv.bn.bias", "backbone.block_list.12.dw_conv.bn.running_mean", "backbone.block_list.12.dw_conv.bn.running_var", "backbone.block_list.12.se.conv1.weight", "backbone.block_list.12.se.conv1.bias", "backbone.block_list.12.se.conv2.weight", "backbone.block_list.12.se.conv2.bias", "backbone.block_list.12.pw_conv.conv.weight", "backbone.block_list.12.pw_conv.bn.weight", "backbone.block_list.12.pw_conv.bn.bias", "backbone.block_list.12.pw_conv.bn.running_mean", "backbone.block_list.12.pw_conv.bn.running_var", "head.ctc_encoder.encoder.conv1.conv.weight", "head.ctc_encoder.encoder.conv1.norm.weight", "head.ctc_encoder.encoder.conv1.norm.bias", "head.ctc_encoder.encoder.conv1.norm.running_mean", "head.ctc_encoder.encoder.conv1.norm.running_var", "head.ctc_encoder.encoder.conv2.conv.weight", "head.ctc_encoder.encoder.conv2.norm.weight", "head.ctc_encoder.encoder.conv2.norm.bias", "head.ctc_encoder.encoder.conv2.norm.running_mean", "head.ctc_encoder.encoder.conv2.norm.running_var", "head.ctc_encoder.encoder.svtr_block.0.norm1.weight", "head.ctc_encoder.encoder.svtr_block.0.norm1.bias", "head.ctc_encoder.encoder.svtr_block.0.mixer.qkv.weight", "head.ctc_encoder.encoder.svtr_block.0.mixer.qkv.bias", "head.ctc_encoder.encoder.svtr_block.0.mixer.proj.weight", "head.ctc_encoder.encoder.svtr_block.0.mixer.proj.bias", "head.ctc_encoder.encoder.svtr_block.0.norm2.weight", "head.ctc_encoder.encoder.svtr_block.0.norm2.bias", "head.ctc_encoder.encoder.svtr_block.0.mlp.fc1.weight", "head.ctc_encoder.encoder.svtr_block.0.mlp.fc1.bias", "head.ctc_encoder.encoder.svtr_block.0.mlp.fc2.weight", "head.ctc_encoder.encoder.svtr_block.0.mlp.fc2.bias", "head.ctc_encoder.encoder.svtr_block.1.norm1.weight", "head.ctc_encoder.encoder.svtr_block.1.norm1.bias", "head.ctc_encoder.encoder.svtr_block.1.mixer.qkv.weight", "head.ctc_encoder.encoder.svtr_block.1.mixer.qkv.bias", "head.ctc_encoder.encoder.svtr_block.1.mixer.proj.weight", "head.ctc_encoder.encoder.svtr_block.1.mixer.proj.bias", "head.ctc_encoder.encoder.svtr_block.1.norm2.weight", "head.ctc_encoder.encoder.svtr_block.1.norm2.bias", "head.ctc_encoder.encoder.svtr_block.1.mlp.fc1.weight", "head.ctc_encoder.encoder.svtr_block.1.mlp.fc1.bias", "head.ctc_encoder.encoder.svtr_block.1.mlp.fc2.weight", "head.ctc_encoder.encoder.svtr_block.1.mlp.fc2.bias", "head.ctc_encoder.encoder.norm.weight", "head.ctc_encoder.encoder.norm.bias", "head.ctc_encoder.encoder.conv3.conv.weight", "head.ctc_encoder.encoder.conv3.norm.weight", "head.ctc_encoder.encoder.conv3.norm.bias", "head.ctc_encoder.encoder.conv3.norm.running_mean", "head.ctc_encoder.encoder.conv3.norm.running_var", "head.ctc_encoder.encoder.conv4.conv.weight", "head.ctc_encoder.encoder.conv4.norm.weight", "head.ctc_encoder.encoder.conv4.norm.bias", "head.ctc_encoder.encoder.conv4.norm.running_mean", "head.ctc_encoder.encoder.conv4.norm.running_var", "head.ctc_encoder.encoder.conv1x1.conv.weight", "head.ctc_encoder.encoder.conv1x1.norm.weight", "head.ctc_encoder.encoder.conv1x1.norm.bias", "head.ctc_encoder.encoder.conv1x1.norm.running_mean", "head.ctc_encoder.encoder.conv1x1.norm.running_var", "head.ctc_head.fc.weight", "head.ctc_head.fc.bias". 

您可以帮忙看看,该怎么改一下能用么? 如果需要模型(训练模型较大,无法上传github),您可以提供一下邮箱,我发给您。

1079863482 commented 2 years ago

1079863482@qq.com,可以发来看下

1079863482 commented 2 years ago

问题在于:因为paddleocr使用了蒸馏训练,ppocrv3的权重里面包含了Teacher和Student的权重,我在转换的函数里通过索引对权重做了个筛选,把前部分Teacher权重过滤了,只保留了后部分的Student权重,因为你的模型没有使用蒸馏,所以不比过滤权重,过滤Teacher权重具体在 if i > 238: 这段代码之后,ppocrV3的权重索引是238以后才是我们需要的。你的模型可以把这部分去掉,可以放开转换函数里的print("{} {}".format(i,key))这部分查看原始权重情况即可明白。

zzzzjx commented 2 years ago

问题在于:因为paddleocr使用了蒸馏训练,ppocrv3的权重里面包含了Teacher和Student的权重,我在转换的函数里通过索引对权重做了个筛选,把前部分Teacher权重过滤了,只保留了后部分的Student权重,因为你的模型没有使用蒸馏,所以不比过滤权重,过滤Teacher权重具体在 if i > 238: 这段代码之后,ppocrV3的权重索引是238以后才是我们需要的。你的模型可以把这部分去掉,可以放开转换函数里的print("{} {}".format(i,key))这部分查看原始权重情况即可明白。

你好,请问转权重的时候过滤掉了teacher的权重,但是我用这个转完之后的权重finetune新的数据集之后,应该权重参数名不会变的,但是在读取权重 的时候还是报错了,和楼上的报错一样,你说的if i > 238:是在那里呢? 谢谢

zzzzjx commented 2 years ago

小哥,你好: 我尝试git下你的源码,转换PPOCRv3官方提供的模型,成功了!但是使用ppocr训练自己的识别模型却转换不成功,报错信息是,缺少很多:如下:

Missing key(s) in state_dict: "backbone.conv1.conv.weight", "backbone.conv1.bn.weight", "backbone.conv1.bn.bias", "backbone.conv1.bn.running_mean", "backbone.conv1.bn.running_var", "backbone.block_list.0.dw_conv.conv.weight", "backbone.block_list.0.dw_conv.bn.weight", "backbone.block_list.0.dw_conv.bn.bias", "backbone.block_list.0.dw_conv.bn.running_mean", "backbone.block_list.0.dw_conv.bn.running_var", "backbone.block_list.0.pw_conv.conv.weight", "backbone.block_list.0.pw_conv.bn.weight", "backbone.block_list.0.pw_conv.bn.bias", "backbone.block_list.0.pw_conv.bn.running_mean", "backbone.block_list.0.pw_conv.bn.running_var", "backbone.block_list.1.dw_conv.conv.weight", "backbone.block_list.1.dw_conv.bn.weight", "backbone.block_list.1.dw_conv.bn.bias", "backbone.block_list.1.dw_conv.bn.running_mean", "backbone.block_list.1.dw_conv.bn.running_var", "backbone.block_list.1.pw_conv.conv.weight", "backbone.block_list.1.pw_conv.bn.weight", "backbone.block_list.1.pw_conv.bn.bias", "backbone.block_list.1.pw_conv.bn.running_mean", "backbone.block_list.1.pw_conv.bn.running_var", "backbone.block_list.2.dw_conv.conv.weight", "backbone.block_list.2.dw_conv.bn.weight", "backbone.block_list.2.dw_conv.bn.bias", "backbone.block_list.2.dw_conv.bn.running_mean", "backbone.block_list.2.dw_conv.bn.running_var", "backbone.block_list.2.pw_conv.conv.weight", "backbone.block_list.2.pw_conv.bn.weight", "backbone.block_list.2.pw_conv.bn.bias", "backbone.block_list.2.pw_conv.bn.running_mean", "backbone.block_list.2.pw_conv.bn.running_var", "backbone.block_list.3.dw_conv.conv.weight", "backbone.block_list.3.dw_conv.bn.weight", "backbone.block_list.3.dw_conv.bn.bias", "backbone.block_list.3.dw_conv.bn.running_mean", "backbone.block_list.3.dw_conv.bn.running_var", "backbone.block_list.3.pw_conv.conv.weight", "backbone.block_list.3.pw_conv.bn.weight", "backbone.block_list.3.pw_conv.bn.bias", "backbone.block_list.3.pw_conv.bn.running_mean", "backbone.block_list.3.pw_conv.bn.running_var", "backbone.block_list.4.dw_conv.conv.weight", "backbone.block_list.4.dw_conv.bn.weight", "backbone.block_list.4.dw_conv.bn.bias", "backbone.block_list.4.dw_conv.bn.running_mean", "backbone.block_list.4.dw_conv.bn.running_var", "backbone.block_list.4.pw_conv.conv.weight", "backbone.block_list.4.pw_conv.bn.weight", "backbone.block_list.4.pw_conv.bn.bias", "backbone.block_list.4.pw_conv.bn.running_mean", "backbone.block_list.4.pw_conv.bn.running_var", "backbone.block_list.5.dw_conv.conv.weight", "backbone.block_list.5.dw_conv.bn.weight", "backbone.block_list.5.dw_conv.bn.bias", "backbone.block_list.5.dw_conv.bn.running_mean", "backbone.block_list.5.dw_conv.bn.running_var", "backbone.block_list.5.pw_conv.conv.weight", "backbone.block_list.5.pw_conv.bn.weight", "backbone.block_list.5.pw_conv.bn.bias", "backbone.block_list.5.pw_conv.bn.running_mean", "backbone.block_list.5.pw_conv.bn.running_var", "backbone.block_list.6.dw_conv.conv.weight", "backbone.block_list.6.dw_conv.bn.weight", "backbone.block_list.6.dw_conv.bn.bias", "backbone.block_list.6.dw_conv.bn.running_mean", "backbone.block_list.6.dw_conv.bn.running_var", "backbone.block_list.6.pw_conv.conv.weight", "backbone.block_list.6.pw_conv.bn.weight", "backbone.block_list.6.pw_conv.bn.bias", "backbone.block_list.6.pw_conv.bn.running_mean", "backbone.block_list.6.pw_conv.bn.running_var", "backbone.block_list.7.dw_conv.conv.weight", "backbone.block_list.7.dw_conv.bn.weight", "backbone.block_list.7.dw_conv.bn.bias", "backbone.block_list.7.dw_conv.bn.running_mean", "backbone.block_list.7.dw_conv.bn.running_var", "backbone.block_list.7.pw_conv.conv.weight", "backbone.block_list.7.pw_conv.bn.weight", "backbone.block_list.7.pw_conv.bn.bias", "backbone.block_list.7.pw_conv.bn.running_mean", "backbone.block_list.7.pw_conv.bn.running_var", "backbone.block_list.8.dw_conv.conv.weight", "backbone.block_list.8.dw_conv.bn.weight", "backbone.block_list.8.dw_conv.bn.bias", "backbone.block_list.8.dw_conv.bn.running_mean", "backbone.block_list.8.dw_conv.bn.running_var", "backbone.block_list.8.pw_conv.conv.weight", "backbone.block_list.8.pw_conv.bn.weight", "backbone.block_list.8.pw_conv.bn.bias", "backbone.block_list.8.pw_conv.bn.running_mean", "backbone.block_list.8.pw_conv.bn.running_var", "backbone.block_list.9.dw_conv.conv.weight", "backbone.block_list.9.dw_conv.bn.weight", "backbone.block_list.9.dw_conv.bn.bias", "backbone.block_list.9.dw_conv.bn.running_mean", "backbone.block_list.9.dw_conv.bn.running_var", "backbone.block_list.9.pw_conv.conv.weight", "backbone.block_list.9.pw_conv.bn.weight", "backbone.block_list.9.pw_conv.bn.bias", "backbone.block_list.9.pw_conv.bn.running_mean", "backbone.block_list.9.pw_conv.bn.running_var", "backbone.block_list.10.dw_conv.conv.weight", "backbone.block_list.10.dw_conv.bn.weight", "backbone.block_list.10.dw_conv.bn.bias", "backbone.block_list.10.dw_conv.bn.running_mean", "backbone.block_list.10.dw_conv.bn.running_var", "backbone.block_list.10.pw_conv.conv.weight", "backbone.block_list.10.pw_conv.bn.weight", "backbone.block_list.10.pw_conv.bn.bias", "backbone.block_list.10.pw_conv.bn.running_mean", "backbone.block_list.10.pw_conv.bn.running_var", "backbone.block_list.11.dw_conv.conv.weight", "backbone.block_list.11.dw_conv.bn.weight", "backbone.block_list.11.dw_conv.bn.bias", "backbone.block_list.11.dw_conv.bn.running_mean", "backbone.block_list.11.dw_conv.bn.running_var", "backbone.block_list.11.se.conv1.weight", "backbone.block_list.11.se.conv1.bias", "backbone.block_list.11.se.conv2.weight", "backbone.block_list.11.se.conv2.bias", "backbone.block_list.11.pw_conv.conv.weight", "backbone.block_list.11.pw_conv.bn.weight", "backbone.block_list.11.pw_conv.bn.bias", "backbone.block_list.11.pw_conv.bn.running_mean", "backbone.block_list.11.pw_conv.bn.running_var", "backbone.block_list.12.dw_conv.conv.weight", "backbone.block_list.12.dw_conv.bn.weight", "backbone.block_list.12.dw_conv.bn.bias", "backbone.block_list.12.dw_conv.bn.running_mean", "backbone.block_list.12.dw_conv.bn.running_var", "backbone.block_list.12.se.conv1.weight", "backbone.block_list.12.se.conv1.bias", "backbone.block_list.12.se.conv2.weight", "backbone.block_list.12.se.conv2.bias", "backbone.block_list.12.pw_conv.conv.weight", "backbone.block_list.12.pw_conv.bn.weight", "backbone.block_list.12.pw_conv.bn.bias", "backbone.block_list.12.pw_conv.bn.running_mean", "backbone.block_list.12.pw_conv.bn.running_var", "head.ctc_encoder.encoder.conv1.conv.weight", "head.ctc_encoder.encoder.conv1.norm.weight", "head.ctc_encoder.encoder.conv1.norm.bias", "head.ctc_encoder.encoder.conv1.norm.running_mean", "head.ctc_encoder.encoder.conv1.norm.running_var", "head.ctc_encoder.encoder.conv2.conv.weight", "head.ctc_encoder.encoder.conv2.norm.weight", "head.ctc_encoder.encoder.conv2.norm.bias", "head.ctc_encoder.encoder.conv2.norm.running_mean", "head.ctc_encoder.encoder.conv2.norm.running_var", "head.ctc_encoder.encoder.svtr_block.0.norm1.weight", "head.ctc_encoder.encoder.svtr_block.0.norm1.bias", "head.ctc_encoder.encoder.svtr_block.0.mixer.qkv.weight", "head.ctc_encoder.encoder.svtr_block.0.mixer.qkv.bias", "head.ctc_encoder.encoder.svtr_block.0.mixer.proj.weight", "head.ctc_encoder.encoder.svtr_block.0.mixer.proj.bias", "head.ctc_encoder.encoder.svtr_block.0.norm2.weight", "head.ctc_encoder.encoder.svtr_block.0.norm2.bias", "head.ctc_encoder.encoder.svtr_block.0.mlp.fc1.weight", "head.ctc_encoder.encoder.svtr_block.0.mlp.fc1.bias", "head.ctc_encoder.encoder.svtr_block.0.mlp.fc2.weight", "head.ctc_encoder.encoder.svtr_block.0.mlp.fc2.bias", "head.ctc_encoder.encoder.svtr_block.1.norm1.weight", "head.ctc_encoder.encoder.svtr_block.1.norm1.bias", "head.ctc_encoder.encoder.svtr_block.1.mixer.qkv.weight", "head.ctc_encoder.encoder.svtr_block.1.mixer.qkv.bias", "head.ctc_encoder.encoder.svtr_block.1.mixer.proj.weight", "head.ctc_encoder.encoder.svtr_block.1.mixer.proj.bias", "head.ctc_encoder.encoder.svtr_block.1.norm2.weight", "head.ctc_encoder.encoder.svtr_block.1.norm2.bias", "head.ctc_encoder.encoder.svtr_block.1.mlp.fc1.weight", "head.ctc_encoder.encoder.svtr_block.1.mlp.fc1.bias", "head.ctc_encoder.encoder.svtr_block.1.mlp.fc2.weight", "head.ctc_encoder.encoder.svtr_block.1.mlp.fc2.bias", "head.ctc_encoder.encoder.norm.weight", "head.ctc_encoder.encoder.norm.bias", "head.ctc_encoder.encoder.conv3.conv.weight", "head.ctc_encoder.encoder.conv3.norm.weight", "head.ctc_encoder.encoder.conv3.norm.bias", "head.ctc_encoder.encoder.conv3.norm.running_mean", "head.ctc_encoder.encoder.conv3.norm.running_var", "head.ctc_encoder.encoder.conv4.conv.weight", "head.ctc_encoder.encoder.conv4.norm.weight", "head.ctc_encoder.encoder.conv4.norm.bias", "head.ctc_encoder.encoder.conv4.norm.running_mean", "head.ctc_encoder.encoder.conv4.norm.running_var", "head.ctc_encoder.encoder.conv1x1.conv.weight", "head.ctc_encoder.encoder.conv1x1.norm.weight", "head.ctc_encoder.encoder.conv1x1.norm.bias", "head.ctc_encoder.encoder.conv1x1.norm.running_mean", "head.ctc_encoder.encoder.conv1x1.norm.running_var", "head.ctc_head.fc.weight", "head.ctc_head.fc.bias". 

您可以帮忙看看,该怎么改一下能用么? 如果需要模型(训练模型较大,无法上传github),您可以提供一下邮箱,我发给您。

请问你解决问题了吗