code: def state_dict(self):
      return {
          n: p.data.cpu()
          for n, p in self.get_trainable_params().items()
      }
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/neox/utils/finetune.py
imports: from typing import List, Dict
  import torch
  from torch import nn
  from labml_nn.neox.model import TransformerLayer, NeoXModule
label_desc: Use image node_id 5 for calling the FineTuner obj's underlying member method code with example usage: obj.state_dict() and returns: unknown