The pretrained weights have some problem

[lingogogo]

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.