14 lines
367 B
YAML
14 lines
367 B
YAML
version: "3"
|
|
tasks:
|
|
- name: "训练小狸AI-Epoch100"
|
|
image:
|
|
dockerfile: .ide/Dockerfile
|
|
resources:
|
|
cpu: 8
|
|
memory: 32G
|
|
gpu: 1
|
|
env:
|
|
CUDA_VISIBLE_DEVICES: "0"
|
|
PYTHONUNBUFFERED: "1"
|
|
commands:
|
|
- cd /workspace && nvidia-smi && echo "=== GPU就绪 ===" && python3 scripts/train_cnb.py --epochs 100 --batch_size 16 |