def forward(self, x):
    out1 = self.linear_patches(self.affine1(x).transpose(1, 2)).transpose(1, 2)
    x = (x + (self.layerscale1 * out1))
    out2 = self.mlp_channels(self.affine2(x))
    x = (x + (self.layerscale2 * out2))
    return x
def forward(self, x):
    out1 = self.linear_patches(self.affine1(x).transpose(1, 2)).transpose(1, 2)
    x = (x + (self.layerscale1 * out1))
    out2 = self.mlp_channels(self.affine2(x))
    x = (x + (self.layerscale2 * out2))
    return x