code: def get_weights(self):
      conv_weights = []
      bn_weights = []
      relu_weights = []
      for pname, p in self.named_parameters():
          if "bn" in pname:
              bn_weights.append(p)
          elif "relu" in pname:
              relu_weights.append(p)
          else:
              conv_weights.append(p)
  
      return conv_weights, bn_weights, relu_weights
  
imports: import os
  import math
  import torch
  import numpy
  import torch.nn
  import torch.nn.functional
  from typing import Optional
  from cftool.cv import to_uint8
  from ....parameters import OPT
label_desc: Use image node_id 2 for calling the PiDiNet obj's underlying member method code with example usage: obj.get_weights() and returns: conv_weights,  bn_weights,  relu_weights