code: def restore(self, checkpoint):
      state_dict = torch.load(checkpoint)
      self.load_state_dict(state_dict)
  
imports: import torch
  from torch.utils.data import TensorDataset, DataLoader
  import types
  import numpy
  import math
  import pandas
  import tempfile
  import os
  from zoo.orca.automl.model.abstract import BaseModel, ModelBuilder
  from zoo.orca.automl.metrics import Evaluator
  from zoo.orca.automl.pytorch_utils import LR_NAME, DEFAULT_LR
label_desc: Use image node_id 20 for calling the PytorchBaseModel obj's underlying member method code with example usage: obj.restore(checkpoint) without return types