code: def make_grad_scaler():
      return flow.amp.GradScaler(
          init_scale=2**30,
          growth_factor=2.0,
          backoff_factor=0.5,
          growth_interval=2000,
      )
  
file_path: insightface/recognition/arcface_oneflow/graph.py
imports: import oneflow
  import oneflow.nn
label_desc: Use image node_id 2 for calling a global function with example usage: make_grad_scaler() and returns: flow