21 lines
519 B
YAML
21 lines
519 B
YAML
$:
|
|
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: 开始训练
|
|
script: |
|
|
cd /workspace
|
|
nvidia-smi
|
|
echo "=== GPU就绪，开始训练 ==="
|
|
python3 scripts/train_cnb.py --epochs 100 --batch_size 16
|
|
echo "=== 训练完成 ===" |