code: def test_load_state_dict_from_torch(self):
      torch.save(model.state_dict(), "resnet18_test.pth")
      pl_model = LightningModule(model, loss, optimizer)
      state_dict = torch.load("resnet18_test.pth")
      pl_model.load_state_dict(state_dict)
  
file_path: BigDL/python/nano/test/pytorch/tests/train/trainer/test_lightning.py
imports: import os
  from unittest import TestCase
  import torch
  import torchmetrics
  from torch import nn
  from test.pytorch.utils._train_torch_lightning import create_data_loader, data_transform
  from bigdl.nano.pytorch.lightning import LightningModule
  from bigdl.nano.pytorch import Trainer
  from bigdl.nano.pytorch.vision.models import vision
label_desc: Use image node_id 2 for calling the TestLightningModule obj's underlying member method code with example usage: obj.test_load_state_dict_from_torch() without return types