Weights do not match network

[delldu]

Dear friend,
Thanks you for sharing this work, paper is good enough, but source code has some issues. We run test.py and found the following erros:
Missing key(s) in state_dict: "module.Tdec.patch_embed1.proj.weight", "module.Tdec.patch_embed1.proj.bias", "module.Tdec.patch_embed1.norm.weight", "module.Tdec.patch_embed1.norm.bias", "module.Tdec.block1.0.norm1.weight", "module.Tdec.block1.0.norm1.bias", "module.Tdec.block1.0.attn.task_query", "module.Tdec.block1.0.attn.q.weight", "module.Tdec.block1.0.attn.q.bias", "module.Tdec.block1.0.attn.kv.weight", "module.Tdec.block1.0.attn.kv.bias", "module.Tdec.block1.0.attn.proj.weight", "module.Tdec.block1.0.attn.proj.bias", "module.Tdec.block1.0.norm2.weight", "module.Tdec.block1.0.norm2.bias", "module.Tdec.block1.0.mlp.fc1.weight", "module.Tdec.block1.0.mlp.fc1.bias", "module.Tdec.block1.0.mlp.dwconv.dwconv.weight", "module.Tdec.block1.0.mlp.dwconv.dwconv.bias", "module.Tdec.block1.0.mlp.fc2.weight", "module.Tdec.block1.0.mlp.fc2.bias", "module.Tdec.block1.1.norm1.weight", "module.Tdec.block1.1.norm1.bias", "module.Tdec.block1.1.attn.task_query", "module.Tdec.block1.1.attn.q.weight", "module.Tdec.block1.1.attn.q.bias", "module.Tdec.block1.1.attn.kv.weight", "module.Tdec.block1.1.attn.kv.bias", "module.Tdec.block1.1.attn.proj.weight", "module.Tdec.block1.1.attn.proj.bias", "module.Tdec.block1.1.norm2.weight", "module.Tdec.block1.1.norm2.bias", "module.Tdec.block1.1.mlp.fc1.weight", "module.Tdec.block1.1.mlp.fc1.bias", "module.Tdec.block1.1.mlp.dwconv.dwconv.weight", "module.Tdec.block1.1.mlp.dwconv.dwconv.bias", "module.Tdec.block1.1.mlp.fc2.weight", "module.Tdec.block1.1.mlp.fc2.bias", "module.Tdec.block1.2.norm1.weight", "module.Tdec.block1.2.norm1.bias", "module.Tdec.block1.2.attn.task_query", "module.Tdec.block1.2.attn.q.weight", "module.Tdec.block1.2.attn.q.bias", "module.Tdec.block1.2.attn.kv.weight", "module.Tdec.block1.2.attn.kv.bias", "module.Tdec.block1.2.attn.proj.weight", "module.Tdec.block1.2.attn.proj.bias", "module.Tdec.block1.2.norm2.weight", "module.Tdec.block1.2.norm2.bias", "module.Tdec.block1.2.mlp.fc1.weight", "module.Tdec.block1.2.mlp.fc1.bias", "module.Tdec.block1.2.mlp.dwconv.dwconv.weight", "module.Tdec.block1.2.mlp.dwconv.dwconv.bias", "module.Tdec.block1.2.mlp.fc2.weight", "module.Tdec.block1.2.mlp.fc2.bias", "module.Tdec.norm1.weight", "module.Tdec.norm1.bias".

It seems weight does not include decoder information.
Would you like check it ?
Thanks in advance.
Best regard,

[lwtgithublwt]

@delldu 我也遇到了，不知道您是否解决，感谢！