code: def get_net(args, criterion):
      net = get_model(
          network=args.arch,
          num_classes=args.dataset_cls.num_classes,
          criterion=criterion,
          trunk=args.trunk,
      )
      num_params = sum([param.nelement() for param in net.parameters()])
      logging.info(
          "Model params = {:2.1f}M".format(num_params / 1000000)
      )
  
      net = net.cuda()
      net = torch.nn.DataParallel(net)
      return net
  
imports: import importlib
  import torch
  import logging
  from .gscnn import GSCNN
label_desc: Use image node_id 1 for calling a global function with example usage: get_net(args, criterion) and returns: net