chore: revert to workspace format (version 3)
This commit is contained in:
1 parent
230d47cade
commit
ac07c7f06a
1 file changed
+14
-23
@@ -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 "=== 训练完成 ==="
|
|
||||||
Reference in new issue
Block a user