Pytorch Upgrade