code: def get_model():
      """Gets the model class specified in the config."""
      err_str = "Model type '{}' not supported"
      assert cfg.MODEL.TYPE in _models.keys(), err_str.format(
          cfg.MODEL.TYPE
      )
      return _models[cfg.MODEL.TYPE]
  
file_path: pycls/pycls/core/builders.py
imports: from pycls.core.config import cfg
  from pycls.core.net import SoftCrossEntropyLoss
  from pycls.models.anynet import AnyNet
  from pycls.models.effnet import EffNet
  from pycls.models.regnet import RegNet
  from pycls.models.resnet import ResNet
  from pycls.models.vit import ViT
label_desc: Use image node_id 1 for calling a global function with example usage: get_model() and returns: _models