code: def get_n_params(model):
      pp = 0
      for p in list(model.parameters()):
          nn = 1
          for s in list(p.size()):
              nn = nn * s
          pp += nn
      return pp
  
imports: import argparse
  import time
  import dgl
  import dgl.function
  import numpy
  import torch
  import torch.nn
  from dataset import load_dataset
label_desc: Use image node_id 1 for calling a global function with example usage: get_n_params(model) and returns: pp