Skip to content

model.load_state_dict error. #12

@Feezhen

Description

@Feezhen

I download the pertrained model, and try to load it. But get the error :
"RuntimeError: Error(s) in loading state_dict for AttModel:
size mismatch for enc_emb.lookup_table: copying a param with shape torch.Size([9786, 512]) from checkpoint, the shape in current model is torch.Size([9796, 512])."

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