chore: revert to workspace format (version 3)

This commit is contained in:
shuiyu1123 committed 2026-06-18 19:22:03 +00:00
1 parent 230d47cade
commit ac07c7f06a
1 file changed
+14 -23
+14 -23
View File
@@ -1,23 +1,14 @@
master: version: "3"
api_trigger: tasks:
- name: 训练小狸AI-Epoch100 - name: "训练小狸AI-Epoch100"
docker: image:
build: .ide/Dockerfile dockerfile: .ide/Dockerfile
runner: resources:
cpus: 8 cpu: 8
tags: cnb:arch:amd64:gpu memory: 32G
env: gpu: 1
CUDA_VISIBLE_DEVICES: "0" env:
PYTHONUNBUFFERED: "1" CUDA_VISIBLE_DEVICES: "0"
git: PYTHONUNBUFFERED: "1"
lfs: false commands:
stages: - cd /workspace && nvidia-smi && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
- name: train
script: |
set -e
echo "=== 小狸AI训练开始 ==="
cd /workspace
nvidia-smi
echo "=== GPU就绪,启动训练 ==="
python3 scripts/train_cnb.py --epochs 100 --batch_size 16
echo "=== 训练完成 ==="