if self.channel_independence == '1': x = self.MLP_channel(x, B, N, T) self.channel_independence == '1' has a problem.