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!