Pytorch Load Module Weights Into 2018