I’m using PyTorch 0.4.1. Since PyTorch 1.0 is a big update, I was afraid of upgrading because of potential incompatibility. However, after a quick search it seems that 0.4.1 code can run seamlessly in 1.0 so I think I will upgrade. Thank a lot!