code: def ensemble(args):
      last = [
          os.path.join(args.exp_dir, args.exp_name, f"epoch={n}.ckpt")
          for n in range(args.epochs - 10, args.epochs)
      ]
      model_path = os.path.join(
          args.exp_dir, args.exp_name, "model_avg_10.pth"
      )
      torch.save({"state_dict": average_checkpoints(last)}, model_path)
  
imports: import os
  import torch
label_desc: Use image node_id 2 for calling a global function with example usage: ensemble(args) without return types