search

szc19990412 / TransMIL

TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image Classification
325 stars 72 forks source link

Float vs. Half #18

Closed 1633146782 closed 1 year ago

1633146782 commented 1 year ago

RuntimeError:expected scalar type Float but found Half

Run train.py files meet this issus,could anybody help me solve it ? thanks

peakhong commented 1 year ago

I have the same error. Have you solved it?

szc19990412 commented 1 year ago

I think this is due to different versions of pytorch lightning framework. Maybe you can try the setting of pytorch_lightning==1.2.3, pytorch_toolbelt==0.4.0.

1633146782 commented 1 year ago

thank you, I will try it again according your advice

------------------ 原始邮件 ------------------ 发件人: @.***>; 发送时间: 2022年10月7日(星期五) 上午10:30 收件人: "szc19990412/TransMIL"; 抄送: "Author"; 主题: Re: [szc19990412/TransMIL] Float vs. Half (Issue #18)

I think this is due to different versions of pytorch lightning framework. Maybe you can try the setting of pytorch_lightning==1.2.3, pytorch_toolbelt==0.4.0.

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>