code: def export(args):
      checkpoint = Checkpoint(
          rank=0,
          world_size=1,
          embedding_size=args.embedding_size,
          num_classes=None,
          checkpoint_dir=args.checkpoint_dir,
      )
  
      backbone = eval("backbones.{}".format(args.backbone))(
          num_features=args.embedding_size
      )
      checkpoint.load(backbone, for_train=False, dtype="float32")
  
      print("Load checkpoint from '{}'.".format(args.checkpoint_dir))
      backbone.eval()
  
      path = os.path.join(args.output_dir, args.backbone)
  
      if args.export_type == "onnx":
          paddle.onnx.export(
              backbone,
              path,
              input_spec=[
                  paddle.static.InputSpec(
                      shape=[None, 3, 112, 112], dtype="float32"
                  )
              ],
          )
      else:
          paddle.jit.save(
              backbone,
              path,
              input_spec=[
                  paddle.static.InputSpec(
                      shape=[None, 3, 112, 112], dtype="float32"
                  )
              ],
          )
      print("Save exported model to '{}'.".format(args.output_dir))
  
imports: import os
  import numpy
  import paddle
  from .utils.io import Checkpoint
  from .None import backbones
label_desc: Use image node_id 1 for calling a global function with example usage: export(args) without return types