Hi @ptrblck, many thanks for your reply. I appreciate it! I could access the plain PyTorch model from Lightning wrapper but I got an error when I tried to load the model.
I got help from another forum and I can load the pretrained model in Lightning now.