code: def get_loss(name):
      if name == "cosface":
          return CosFace()
      elif name == "arcface":
          return ArcFace()
      else:
          raise ValueError()
  
file_path: insightface/recognition/arcface_oneflow/utils/losses.py
imports: import oneflow
  from oneflow import nn
label_desc: Use image node_id 2 for calling a global function with example usage: get_loss(name) and returns: CosFace, ArcFace