How can I load torch model into pytorch?