Couldn't visualize the captions generated on IU_Xray dataset by revised the plot_mimic_cxr.sh.

[RunCode90]

Hi, good work and thank you!

I followed the bash plot_mimic_cxr.sh step you have mentioned. I want to visualize the captions generated on IU_Xray dataset by using model_iu_xray.pth, since i am not yet able to download the MIMIC_CXR dataset.

But it has the following errors：

RuntimeError: Error(s) in loading state_dict for R2GenModel:
size mismatch for encoder_decoder.model.tgt_embed.0.lut.weight: copying a param with shape torch.Size([761, 512]) from checkpoint, the shape in current model is torch.Size([400, 512]).
size mismatch for encoder_decoder.logit.weight: copying a param with shape torch.Size([761, 512]) from checkpoint, the shape in current model is torch.Size([400, 512]).
size mismatch for encoder_decoder.logit.bias: copying a param with shape torch.Size([761]) from checkpoint, the shape in current model is torch.Size([400]).

Kindly any guidance， thanks!