mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-04-22 18:13:52 +00:00
chore: Regenerate all playbooks
This commit is contained in:
parent
3a001a2e76
commit
b8516f3d69
@ -73,7 +73,7 @@ def main(args):
|
||||
# Configure the SFT config
|
||||
config = {
|
||||
"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,
|
||||
"learning_rate": args.learning_rate,
|
||||
"optim": "adamw_torch",
|
||||
|
||||
Loading…
Reference in New Issue
Block a user