diff --git a/.cnb.yml b/.cnb.yml
index e3fd839..e51a4b4 100644
--- a/.cnb.yml
+++ b/.cnb.yml
@@ -1,14 +1,18 @@
-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
+$:
+  vscode:
+    - docker:
+        build: .ide/Dockerfile
+      runner:
+        cpus: 8
+        tags: cnb:arch:amd64:gpu
+      services:
+        - vscode
+        - docker
+      env:
+        CUDA_VISIBLE_DEVICES: "0"
+        PYTHONUNBUFFERED: "1"
+      stages:
+        - name: ls
+          script: ls -al
+        - name: 开始训练
+          script: cd /workspace && nvidia-smi && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
\ No newline at end of file