Pytorch Load Module Weights Into 2 Graphs