It seems you’ve removed the model.eval() call, which would disable dropout and use the running stats in batchnorm layers instead of the sample statistics.
Could you call model.eval() again and check the results?
Also, how large are the differences between the runs?