code: def get_AFN(num_classes=1, arch="AFN4", **kwargs):
      return ModelBuilder().build_net(
          arch=arch, num_classes=num_classes, **kwargs
      )
  
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: get_AFN(num_classes, arch) and returns: ModelBuilder