GLMix
GLMix copied to clipboard
您好,代码GLMixBlock类中的self.norm1和self.norm2执行时会报错
这里输入的x形状为[batch,c,h,w],但是self.norm1是LayerNorm2d((64, 1, 1), eps=1e-05, elementwise_affine=True),执行的时候在维度上会报错。self.norm2也是这个问题。