code: def ShuffleNetG2():
      cfg = {
          "out_planes": [200, 400, 800],
          "num_blocks": [4, 8, 4],
          "groups": 2,
      }
      return ShuffleNet(cfg)
  
file_path: nni/examples/trials/cifar10_pytorch/models/shufflenet.py
imports: import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 for calling a global function with example usage: ShuffleNetG2() and returns: ShuffleNet