Pytorch Load Module Weights Into 2 Graphs That Compare