feat: add CNB cloud training config (GPU Dockerfile + .cnb.yml + train_cnb.py)
This commit is contained in:
1 parent
eb7717e9ce
commit
456c443e9c
4 files changed
+223
-1
No files matched your search
@@ -0,0 +1,15 @@
|
||||
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 50 --batch_size 16
|
||||
Reference in new issue
Block a user