Pytorch Load State Dict Code