madroid
/

tinyllama-flex-lora

Generated from Trainer

Model card Files Files and versions Community

madroid commited on Oct 13, 2023

Commit

21759ac

·

1 Parent(s): 10468fc

End of training

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -33,11 +33,11 @@ More information needed
 The following hyperparameters were used during training:
 - learning_rate: 0.0002
-- train_batch_size: 8
 - eval_batch_size: 8
 - seed: 42
 - gradient_accumulation_steps: 4
-- total_train_batch_size: 32
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - num_epochs: 10

 The following hyperparameters were used during training:
 - learning_rate: 0.0002
+- train_batch_size: 18
 - eval_batch_size: 8
 - seed: 42
 - gradient_accumulation_steps: 4
+- total_train_batch_size: 72
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - num_epochs: 10