-
Notifications
You must be signed in to change notification settings - Fork 1.7k
Description
问题确认 Search before asking
- 我已经查询历史issue(包括open与closed),没有发现相似的bug。I have searched the open and closed issues and found no similar bug report.
Bug描述 Describe the Bug
ppliteseg加载预训练模型 后报[2025/12/03 09:49:26] WARNING: smb1.blocks.0.linear_conv.bn._variance is not in pretrained model的错误,随后我打印训练好的模型文件中model.state_dict()的key,发现打印的key是backbone.smb1.blocks.0.linear_conv.bn._variance, 两个key对应不上,导致警告。
复现环境 Environment
[2025/12/03 09:49:25] INFO: Set device: gpu
[2025/12/03 09:49:25] INFO: Use the following config to build model
model:
backbone:
inj_type: AAM
out_feat_chs:
- 32
- 64
- 128
pretrained: ./output/moqu1202/best_model/model.pdparams
type: MobileSeg_Tiny
num_classes: 12
type: PPMobileSeg
upsample: intepolate
W1203 09:49:25.058808 415337 gpu_resources.cc:119] Please NOTE: device: 1, GPU Compute Capability: 8.6, Driver API Version: 12.9, Runtime API Version: 11.8
W1203 09:49:25.059067 415337 gpu_resources.cc:164] device: 1, cuDNN Version: 8.7.
[2025/12/03 09:49:26] INFO: Loading pretrained model from ./output/moqu1202/best_model/model.pdparams
[2025/12/03 09:49:26] WARNING: smb1.conv.c.weight is not in pretrained model
[2025/12/03 09:49:26] WARNING: smb1.conv.bn.weight is not in pretrained model
[2025/12/03 09:49:26] WARNING: smb1.conv.bn.bias is not in pretrained model
[2025/12/03 09:49:26] WARNING: smb1.conv.bn._mean is not in pretrained model
Bug描述确认 Bug description confirmation
- 我确认已经提供了Bug复现步骤、代码改动说明、以及环境信息,确认问题是可以复现的。I confirm that the bug replication steps, code change instructions, and environment information have been provided, and the problem can be reproduced.
是否愿意提交PR? Are you willing to submit a PR?
- 我愿意提交PR!I'd like to help by submitting a PR!