cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add --check-only flag to validate setup without training

Browse Source
This commit is contained in:
Christian Medina
2026-06-30 20:39:29 -04:00
parent 83ba987840
commit 6f31fa0197
2 changed files with 85 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train-on-this-server.sh
View File

@@ -78,6 +78,8 @@ echo "Press Ctrl+C to stop (model will be saved at checkpoint)."
echo ""
# Use torchrun for distributed training (2 GPUs)
# Add --check-only to validate without training:
# torchrun --nproc_per_node=2 training/scripts/train.py --config training/configs/llama2-7b-lora.yaml --check-only
torchrun --nproc_per_node=2 training/scripts/train.py --config training/configs/llama2-7b-lora.yaml
echo ""