code: def init_weights(self):
      for m in self.modules():
          if isinstance(m, nn.Conv3d):
              pass
          if isinstance(m, nn.BatchNorm3d):
              m.weight.data.fill_(1)
              m.bias.data.fill_(0)
  
      if self.aux_head is not None:
          self.aux_head.init_weights()
  
imports: import numpy
  import torch
  import torch.nn
  import torch.utils.checkpoint
  import torch.nn.functional
label_desc: Use image node_id 2 for calling the TPN obj's underlying member method code with example usage: obj.init_weights() without return types