version: "3"
tasks:
  - name: "训练小狸AI-Epoch100"
    image:
      dockerfile: .ide/Dockerfile
    resources:
      cpu: 8
      memory: 32G
      gpu: 1
    env:
      CUDA_VISIBLE_DEVICES: "0"
      PYTHONUNBUFFERED: "1"
    commands:
      - cd /workspace && nvidia-smi && echo "=== GPU就绪 ===" && python3 scripts/train_cnb.py --epochs 100 --batch_size 16