code: def get_trainable_params(self) -> Dict[str, nn.Parameter]:
      params = {}
      for i, layer in enumerate(self.layers):
          params.update(
              self.get_layer_trainable_params(
                  layer, prefix=f"layer_{i :02d}"
              )
          )
  
      return params
  
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 2 for calling the FineTuner obj's underlying member method code with example usage: obj.get_trainable_params() and returns: params