Skip to content

KeyError: 'missing keys in state_dict: "{\'body_t.bias\', \'body_t.weight\'}"' When loading the pre-trained weights. #2

@ANYMS-A

Description

@ANYMS-A

It seems the pretrianed weights doesn't contain the weights of the self.body_t layer's weights and bias. So is this layer is removed from the model when training the RLKA-Net?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions