xh-liu / CC-FPSE

Code for NeurIPS 2019 paper "Learning to Predict Layout-to-image Conditional Convolutions for Semantic Image Synthesis"
128 stars 15 forks source link

The pretrained weights have some problem #14

Open lingogogo opened 7 months ago

lingogogo commented 7 months ago

Hello, we encounter some problem with your code about the paper.

The following is our problem.

image

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.