Depending on the batch size (and potentially other hyperparameters you’ve changed), the output might not be exactly the same.
I would recommend to update to the latest stable PyTorch release (1.5.0) and rerun the script again.
0.4.1 is quite old by now and a lot of fixes went into the code since then.