diff --git a/.cnb.yml b/.cnb.yml
index f09fc2b..e3fd839 100644
--- a/.cnb.yml
+++ b/.cnb.yml
@@ -1,23 +1,14 @@
-master:
-  api_trigger:
-    - name: 训练小狸AI-Epoch100
-      docker:
-        build: .ide/Dockerfile
-      runner:
-        cpus: 8
-        tags: cnb:arch:amd64:gpu
-      env:
-        CUDA_VISIBLE_DEVICES: "0"
-        PYTHONUNBUFFERED: "1"
-      git:
-        lfs: false
-      stages:
-        - name: train
-          script: |
-            set -e
-            echo "=== 小狸AI训练开始 ==="
-            cd /workspace
-            nvidia-smi
-            echo "=== GPU就绪,启动训练 ==="
-            python3 scripts/train_cnb.py --epochs 100 --batch_size 16
-            echo "=== 训练完成 ==="
\ No newline at end of file
+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 && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
\ No newline at end of file