RuntimeError: Expected target size [16, 18], got [16, 14]

[newfish-lab]

作者你好！我训练的时候碰到报错“RuntimeError: Expected target size [16, 18], got [16, 14]”，论文提到disease labels是14，我想知道是不是代码哪里有问题？谢谢作者！

[jhb86253817]

论文中提到过，除了14个标准疾病，我们还增加了4个额外的辅助疾病，我们提供的json文件中包含了这18个label。请再确认一下是否使用了我们提供的标注。