16 lines
474 B
YAML
16 lines
474 B
YAML
master:
|
|
api_trigger:
|
|
- name: 训练小狸AI-Epoch100
|
|
docker:
|
|
build: .ide/Dockerfile
|
|
runner:
|
|
cpus: 8
|
|
tags: cnb:arch:amd64:gpu
|
|
env:
|
|
CUDA_VISIBLE_DEVICES: "0"
|
|
PYTHONUNBUFFERED: "1"
|
|
stages:
|
|
- name: train
|
|
script: |
|
|
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 |