code: def main(args):
      cfg = get_config(args.config)
      cfg.graph = args.graph
      rank = flow.env.get_rank()
      world_size = flow.env.get_world_size()
      placement = flow.env.all_device_placement("cuda")
  
      os.makedirs(cfg.output, exist_ok=True)
      log_root = logging.getLogger()
      init_logging(log_root, rank, cfg.output)
  
      # root dir of loading checkpoint
      load_path = None
  
      for key, value in cfg.items():
          num_space = 25 - len(key)
          logging.info(": " + key + " " * num_space + str(value))
  
      trainer = Trainer(cfg, placement, load_path, world_size, rank)
      trainer()
  
file_path: insightface/recognition/arcface_oneflow/train.py
imports: import argparse
  import logging
  import os
  import oneflow
  from function import Trainer
  from utils.utils_logging import init_logging
  from utils.utils_config import get_config
label_desc: Use image node_id 1 for calling a global function with example usage: main(args) without return types