您好，代码GLMixBlock类中的self.norm1和self.norm2执行时会报错

[Run-19]

这里输入的x形状为[batch,c,h,w]，但是self.norm1是LayerNorm2d((64, 1, 1), eps=1e-05, elementwise_affine=True)，执行的时候在维度上会报错。self.norm2也是这个问题。

[rayleizhu]

检查一下环境依赖？我发布前运行过推理，没有遇到问题的 On 20 Dec 2024, at 3:04 PM, ZhengRun ***@***.***> wrote: image.png (view on web)<https://github.com/user-attachments/assets/27d75713-a4ae-45ad-a61d-dd2a75616b41> 这里输入的x形状为[batch,c,h,w]，但是self.norm1是LayerNorm2d((64, 1, 1), eps=1e-05, elementwise_affine=True)，执行的时候在维度上会报错。self.norm2也是这个问题。 — Reply to this email directly, view it on GitHub<#1>, or unsubscribe<https://github.com/notifications/unsubscribe-auth/AEYCTO37RTDIUMHGDA6E4MD2GO6RDAVCNFSM6AAAAABT6RC4ZGVHI2DSMVQWIX3LMV43ASLTON2WKOZSG42TEMBYGYYDCNI>. You are receiving this because you are subscribed to this thread.Message ID: ***@***.***>