chore: rewrite .cnb.yml to CI pipeline format with api_trigger event
This commit is contained in:
1 parent
9799aa00c5
commit
edf3a83e3c
1 file changed
+16
-15
@@ -1,15 +1,16 @@
|
|||||||
version: "3"
|
master:
|
||||||
tasks:
|
api_trigger:
|
||||||
- name: "训练小狸AI"
|
- name: 训练小狸AI-Epoch100
|
||||||
image:
|
docker:
|
||||||
dockerfile: .ide/Dockerfile
|
build: .ide/Dockerfile
|
||||||
resources:
|
runner:
|
||||||
cpu: 8
|
cpus: 8
|
||||||
memory: 32G
|
tags: cnb:arch:amd64:gpu
|
||||||
gpu: 1
|
env:
|
||||||
env:
|
CUDA_VISIBLE_DEVICES: "0"
|
||||||
CUDA_VISIBLE_DEVICES: "0"
|
PYTHONUNBUFFERED: "1"
|
||||||
PYTHONUNBUFFERED: "1"
|
stages:
|
||||||
commands:
|
- name: train
|
||||||
- cd /workspace && git clone https://cnb.cool/ETeam_sy/xiaoli-llm-chat.git
|
script: |
|
||||||
- cd xiaoli-llm-chat && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
|
cd /workspace && git clone https://cnb.cool/ETeam_sy/xiaoli-llm-chat.git
|
||||||
|
cd xiaoli-llm-chat && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
|
||||||
Reference in new issue
Block a user