@mrshenli I don’t think this is related to the view/inplace fix that is mentioned there.
There are no views involved in this example and the error is a “classic” change inplace of a Tensor that is needed for gradients. Since the failing Node is batchnorm, I guess that that would either be in the input to the Module or the weights of the affine transformation.
Is DDP modifying weights inplace by any chance? Or the optimizer is not used properly?