Pytorch get model weights.