Skip to content

size mismatch for decoder #76

Description

@mrub1

Dear Authors,
Thank you so much for a great paper !
Did a training on 1 shapenet class.
Trying generate.py on this newly trained model.
Getting:
RuntimeError: Error(s) in loading state_dict for DVR:
size mismatch for decoder.fc_p.weight: copying a param with shape torch.Size([128, 3]) from checkpoint, the shape in current model is torch.Size([512, 3]).
Going through the code to identify which parameter should be adjusted.
Any hints would be appreciated.
No such issue on the pretrained model
Thanks a lot !

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions