missing tfc module

[xkyi1111]

Thank you for sharing your excellent work; I've run into an issue and would appreciate your help.

	Missing key(s) in state_dict: "CRM_7.conv_p1.weight", "CRM_7.conv_p1.bias", "CRM_7.conv_p2.weight", "CRM_7.conv_p2.bias", "CRM_7.conv_glb.weight", "CRM_7.conv_glb.bias", "CRM_7.conv_matt.0.weight", "CRM_7.conv_matt.0.bias", "CRM_7.conv_matt.1.weight", "CRM_7.conv_matt.1.bias", "CRM_7.conv_matt.1.running_mean", "CRM_7.conv_matt.1.running_var", "CRM_7.conv_fuse.0.weight", "CRM_7.conv_fuse.0.bias", "CRM_7.conv_fuse.1.weight", 
...
	Unexpected key(s) in state_dict: "tfc_7.conv.weight", "tfc_7.conv.bias", "tfc_7.conv2.weight", "tfc_7.conv2.bias", "tfc_7.convss.weight", "tfc_7.convss.bias", "tfc_7.conv_b.0.weight", "tfc_7.conv_b.0.bias", "tfc_7.conv_b.1.weight", "tfc_7.conv_b.1.bias", "tfc_7.conv_b.1.running_mean", "tfc_7.conv_b.1.running_var", "tfc_7.conv_b.1.num_batches_tracked", "tfc_7.conv_ss.0.weight", "tfc_7.conv_ss.0.bias", "tfc_7.conv_ss.1.weight", "tfc_7.conv_ss.1.bias", "tfc_7.conv_ss.1.running_mean", "tfc_7.conv_ss.1.running_var", "tfc_7.conv_ss.1.num_batches_tracked", "tfc_7.conv_ss.3.weight", "tfc_7.conv_ss.3.bias", "tfc_7.conv_ss.4.weight", "tfc_7.conv_ss.4.bias", "tfc_7.conv_ss.4.running_mean", "tfc_7.conv_ss.4.running_var", .
...

[BarCodeReader]

did you manage to load the checkpoint correctly?
for the main branch, you need to load pvt backbone, while if you want to use Swin, please pull Swin branch.

[maoyj2333]

@BarCodeReader hello, I also encounter this issue, i found that:

1. keys in 'best_DUTS-TE_Swin.pt' do not contain 'CRM_7' or any other CRM module

2. in branch 'swin-sod' - 'network_swin.py' - line 57, there is a claim about of multiple CRM mdoule.

[BarCodeReader]

ok let me check tonight

[BarCodeReader]

i fixed some minor issue and tested, it works fine. i believe you did not checkout to swin branch