Error while training on CIFAR!!

[yutinyang]

raise EinopsError('Shape mismatch, {} != {}'.format(length, known_product))
einops.EinopsError: Shape mismatch, 4 != 196

[rishikksh20]

@yutinyang error occur in which line number ?

[yutinyang]

 line 24, in forward     x = leff(x[:, 1:]) 

…

------------------ 原始邮件 ------------------ 发件人: "rishikksh20/CeiT-pytorch" ***@***.***>; 发送时间: 2021年4月9日(星期五) 下午5:31 ***@***.***>; ***@***.******@***.***>; 主题: Re: [rishikksh20/CeiT-pytorch] Error while training on CIFAR!! (#1) @yutinyang error occur in which line number ? — You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub, or unsubscribe.

[khawar-islam]

@rishikksh20 @yutinyang I am also following the same error

[iwanglei1]

The error was caused by the TorchVision version. Install Torchvision ==0.9 Torch ==1.8.1 to solve the problem.

[yuwanting828]

I also following the same error. Can you help me?Thank you!