首页 Paddle框架 帖子详情
PPLiteSeg 使用预训练模型问题
收藏
快速回复
Paddle框架 问答模型训练深度学习 297 1
PPLiteSeg 使用预训练模型问题
收藏
快速回复
Paddle框架 问答模型训练深度学习 297 1

PPLiteSeg 使用自己训练的预训练模型,无法加载参数进去

config文件:

model:
  type: PPLiteSeg
  backbone:
    type: STDC1
    pretrained: /home/aistudio/PaddleSeg-develop/output/iter_10/model.pdparams

开始训练:

2023-02-03 10:55:46 [INFO] Loading pretrained model from /home/aistudio/PaddleSeg-develop/output/iter_10/model.pdparams
2023-02-03 10:55:46 [WARNING] features.0.conv.weight is not in pretrained model
2023-02-03 10:55:46 [WARNING] features.0.bn.weight is not in pretrained model
2023-02-03 10:55:46 [WARNING] features.0.bn.bias is not in pretrained model
2023-02-03 10:55:46 [WARNING] features.0.bn._mean is not in pretrained model
2023-02-03 10:55:46 [WARNING] features.0.bn._variance is not in pretrained model
...................................................省略N行.................................................................
2023-02-03 10:55:46 [WARNING] x32.0.1.conv_list.3.bn._mean is not in pretrained model
2023-02-03 10:55:46 [WARNING] x32.0.1.conv_list.3.bn._variance is not in pretrained model
2023-02-03 10:55:46 [INFO] There are 0/295 variables loaded into STDCNet.

0
收藏
回复
全部评论(1)
时间顺序
宇宙物语
#2 回复于2023-02

建议检查一下自己训练的模型和加载的模型结构是否相同

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户