code: def num_params(module):
      return sum(
          p.numel() for p in module.parameters() if p.requires_grad
      )
  
file_path: tensorwatch/tensorwatch/model_graph/torchstat/compute_memory.py
imports: import torch.nn
  import torch
  import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: num_params(module) and returns: sum