AttnGAN icon indicating copy to clipboard operation
AttnGAN copied to clipboard

RuntimeError: Error(s) in loading state_dict for RNN_ENCODER:

Open hs457681503 opened this issue 4 years ago • 1 comments

size mismatch for encoder.weight: copying a param of torch.Size([1, 300]) from checkpoint, where the shape is torch.Size([5450, 300]) in current model.

hs457681503 avatar Jul 24 '20 11:07 hs457681503

Look at your birds dataset's caption. Pickle if it's 6M it will run and 4M it will report an error. Here's how mine worked out

hannanyi avatar Sep 22 '22 11:09 hannanyi