code: def main():
      model = YOLO("yolov8n.pt")
      add_wandb_callback(
          model,
          max_validation_batches=2,
          enable_model_checkpointing=True,
      )
      model.train(data="coco128.yaml", epochs=2, imgsz=640, batch=64)
      model.val()
  
label_desc: Use image node_id 1 for calling a global function with example usage: main() without return types