Load models

[wsw321]

I add some layers in osnet_ain.py/OSNet, and I am train successfully. When I am trying to load weights on model using function from the tool that giving me

The following layers are discarded due to unmatched keys or layer size: ['classifier.weight', 'classifier.bias', 'conv4_local.0.conv1.conv.weight', 'conv4_local.0.conv1.bn.weight', 'conv4_local.0.conv1.bn.bias',...]

It seems that the layers I added are not loaded. How can I solve this problem? Thank you very much.

[J-LINC]

i have the same question!

[J-LINC]

I add some layers in osnet_ain.py/OSNet, and I am train successfully. When I am trying to load weights on model using function from the tool that giving me

The following layers are discarded due to unmatched keys or layer size: ['classifier.weight', 'classifier.bias', 'conv4_local.0.conv1.conv.weight', 'conv4_local.0.conv1.bn.weight', 'conv4_local.0.conv1.bn.bias',...]

It seems that the layers I added are not loaded. How can I solve this problem? Thank you very much.

i have solve this question, The content is as follows

[wsw321]

I add some layers in osnet_ain.py/OSNet, and I am train successfully. When I am trying to load weights on model using function from the tool that giving me The following layers are discarded due to unmatched keys or layer size: ['classifier.weight', 'classifier.bias', 'conv4_local.0.conv1.conv.weight', 'conv4_local.0.conv1.bn.weight', 'conv4_local.0.conv1.bn.bias',...] It seems that the layers I added are not loaded. How can I solve this problem? Thank you very much.

i have solve this question, The content is as follows

Wow, incredible! How did you solve it? Where do you need to modify the code, thank you!

[J-LINC]

okay ,emmm，Firstly, this is the code I want to run

and Here is the model I downloaded

run the code ,i meet :

[J-LINC]

after that, i do some changes: then i successed!

[J-LINC]

I add some layers in osnet_ain.py/OSNet, and I am train successfully. When I am trying to load weights on model using function from the tool that giving me The following layers are discarded due to unmatched keys or layer size: ['classifier.weight', 'classifier.bias', 'conv4_local.0.conv1.conv.weight', 'conv4_local.0.conv1.bn.weight', 'conv4_local.0.conv1.bn.bias',...] It seems that the layers I added are not loaded. How can I solve this problem? Thank you very much.

i have solve this question, The content is as follows

Look at this prompt： “Incompatible layers (unmatched in name or size) will be ignored.”

[wsw321]

I add some layers in osnet_ain.py/OSNet, and I am train successfully. When I am trying to load weights on model using function from the tool that giving me The following layers are discarded due to unmatched keys or layer size: ['classifier.weight', 'classifier.bias', 'conv4_local.0.conv1.conv.weight', 'conv4_local.0.conv1.bn.weight', 'conv4_local.0.conv1.bn.bias',...] It seems that the layers I added are not loaded. How can I solve this problem? Thank you very much.

i have solve this question, The content is as follows

Look at this prompt： “Incompatible layers (unmatched in name or size) will be ignored.”

Okay, your suggestions are very helpful to me. Thank you for your reply.