The pretrained weights have some problem
Opened this issue · 0 comments
lingogogo commented
Hello, we encounter some problem with your code about the paper.
The following is our problem.
The error is below:
The size mismatch for labelenc1.0.0.weight_orig: copying a param with shape torch.Size([64, 184, 3, 3]) from checkpoint, the shape in current model is torch.Size([64, 183, 3, 3]).
test_coco.sh has already use the argument --use_vae
Thank you so much.