i\'m trying to work out this inplase error, it seems unable to compute the gradient in the last linear layer. I think this is due to the previous global max pooling layer. B
