search

zhjohnchan / R2GenCMN

[ACL-2021] The official implementation of Cross-modal Memory Networks for Radiology Report Generation.
Apache License 2.0
77 stars 7 forks source link

IU X-Ray提供的权重和模型的对齐问题 #8

Closed Liqq1 closed 1 year ago

Liqq1 commented 1 year ago

权重文件为Readme中下载的model_iu_xray.pth 运行bash test_iu_xray.sh 会有以下报错：

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([760, 512]).

同时还出现大量的Missing key(s) in state_dict

zhjohnchan commented 1 year ago

Hi @Liqq1,

Thanks for your attention! Please check if the checkpoint is downloaded correctly.

Best, Zhihong