Pytorch Lightning Load Pretrained Model