Backbone部分没有载入是错误的。

[molywyp]

PS D:\segformer-pytorch> & d:/anaconda/python.exe d:/segformer-pytorch/train.py
Load backbone weights
Load weights model_data\segformer_b5_backbone_weights.pth.

Successful Load Key: [] ……
Successful Load Key Num: 0

Fail To Load Key: ['patch_embed1.proj.weight', 'patch_embed1.proj.bias', 'patch_embed1.norm.weight', 'patch_embed1.norm.bias', 'patch_embed2.proj.weight', 'patch_embed2.proj.bias', 'patch_embed2.norm.weight', 'patch_embed2.norm.bias', 'patch_embed3.proj.weight', 'patch_embed3.proj.bias', 'patch_embed3.norm.weight', 'patch_embed3.norm.bias', 'patch_embed4.proj.weight', 'patch_embed4.proj.bias', 'patch_embed4.norm.weight', 'patch_embed4.norm.bias', 'block1.0.norm1.weight', 'block1.0.norm1.bias', 'block1.0.attn.q.we ……
Fail To Load Key num: 1054

温馨提示，head部分没有载入是正常现象，Backbone部分没有载入是错误的。
Configurations:

| keys | values|

| num_classes | 21|
| phi | b0|
| model_path | model_data\segformer_b5_backbone_weights.pth|
| input_shape | [512, 512]|
| Init_Epoch | 0|
| Freeze_Epoch | 50|
| UnFreeze_Epoch | 100|
| Freeze_batch_size | 16|
| Unfreeze_batch_size | 8|
| Freeze_Train | True|
| Init_lr | 0.0001|
| Min_lr | 1.0000000000000002e-06|
| optimizer_type | adamw|
| momentum | 0.9|
| lr_decay_type | cos|
| save_period | 5|
| save_dir | logs|
| num_workers | 4|
| num_train | 10582|
| num_val | 1449|

[sym330]

+1