chore: revert to version 3 tasks format for SSH support

This commit is contained in:
shuiyu1123 committed 2026-06-18 20:05:41 +00:00
1 parent f5dd3e5874
commit c6ff7be55a
1 file changed
+14 -18
+14 -18
View File
@@ -1,18 +1,14 @@
$: version: "3"
vscode: tasks:
- docker: - name: "训练小狸AI-Epoch100"
build: .ide/Dockerfile image:
runner: dockerfile: .ide/Dockerfile
cpus: 8 resources:
tags: cnb:arch:amd64:gpu cpu: 8
services: memory: 32G
- vscode gpu: 1
- docker env:
env: CUDA_VISIBLE_DEVICES: "0"
CUDA_VISIBLE_DEVICES: "0" PYTHONUNBUFFERED: "1"
PYTHONUNBUFFERED: "1" commands:
stages: - cd /workspace && nvidia-smi && echo "=== GPU就绪 ===" && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
- name: ls
script: ls -al
- name: 开始训练
script: cd /workspace && nvidia-smi && python3 scripts/train_cnb.py --epochs 100 --batch_size 16