I try but the problem is only for my output layer.If i’m going to change i the input i have this error
x = torch.randn(1,3, 224, 224)
x = x.permute(0, 3, 1, 2)

RuntimeError: Given groups=1, weight of size 32 3 3 3, expected input[1, 224, 4, 225] to have 3 channels, but got 224 channels instead