code: def get_model_from_arch_str(
      arch_str, num_classes, bn=True, init_channels=16
  ):
      keep_mask = gen_searchcell_mask_from_arch_str(arch_str)
      net = NAS201Model(
          arch_str=arch_str,
          num_classes=num_classes,
          use_bn=bn,
          keep_mask=keep_mask,
          stem_ch=init_channels,
      )
      return net
  
file_path: singa/examples/model_selection/TRAILS-Database-Native-Model-Selection/internal/ml/model_selection/src/third_pkg/sp201_lib/nasbench2.py
imports: from .nasbench2_ops import *
label_desc: Use image node_id 2 for calling a global function with example usage: get_model_from_arch_str(arch_str, num_classes, bn, init_channels) and returns: net