size mismatch and AssertionError

[xmdgaoxin]

Hello, I have the following problems during training, how should I solve them

[joeyz0z]

I meet the same problem, do you solve it?

[xmdgaoxin]

I remember that the problem was solved after modifying num_class in the definition file of vit, hope this will help you. 907218307 ***@***.***

…

------------------ 原始邮件 ------------------ 发件人: "ViTAE-Transformer/ViTDet" ***@***.***>; 发送时间: 2023年6月8日(星期四) 中午11:51 ***@***.***>; ***@***.******@***.***>; 主题: Re: [ViTAE-Transformer/ViTDet] size mismatch and AssertionError (Issue #22) I meet the same problem, do you solve it? — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>

[joeyz0z]

Thanks for your prompt suggestion, it helps me a lot!