sarman/dgx-spark-playbooks
1
0
Fork 0
mirror of https://github.com/NVIDIA/dgx-spark-playbooks.git synced 2026-04-26 11:53:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: Regenerate all playbooks

Browse Source
This commit is contained in:
GitLab CI 2026-01-16 18:33:03 +00:00
parent 3a001a2e76
commit b8516f3d69
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nvidia/pytorch-fine-tune/assets/Llama3_8B_LoRA_finetuning.py
View File

@ -73,7 +73,7 @@ def main(args):
# Configure the SFT config # Configure the SFT config
config = { config = {
"per_device_train_batch_size": args.batch_size, "per_device_train_batch_size": args.batch_size,
"num_train_epochs": 0.01, "num_train_epochs": 0.05,
"gradient_accumulation_steps": args.gradient_accumulation_steps, "gradient_accumulation_steps": args.gradient_accumulation_steps,
"learning_rate": args.learning_rate, "learning_rate": args.learning_rate,
"optim": "adamw_torch", "optim": "adamw_torch",