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.
建议检查一下自己训练的模型和加载的模型结构是否相同