How to load the pretrained weights into the corresponding model?