15 lines
402 B
YAML
15 lines
402 B
YAML
version: "3"
|
|
tasks:
|
|
- name: "训练小狸AI"
|
|
image:
|
|
dockerfile: .ide/Dockerfile
|
|
resources:
|
|
cpu: 8
|
|
memory: 32G
|
|
gpu: 1
|
|
env:
|
|
CUDA_VISIBLE_DEVICES: "0"
|
|
PYTHONUNBUFFERED: "1"
|
|
commands:
|
|
- 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 |