As @imaluengo explained, both approaches should yield the same results.
However, you would have to be careful, if your model contains any nn.BatchNorm layers, as the running estimates will be different using these approaches.