Upload sft_gen.yaml with huggingface_hub
Browse files- sft_gen.yaml +1 -1
sft_gen.yaml
CHANGED
|
@@ -27,7 +27,7 @@ train_cfgs:
|
|
| 27 |
# Seed for random number generator
|
| 28 |
seed: 42
|
| 29 |
# Batch size per device for training
|
| 30 |
-
per_device_train_batch_size:
|
| 31 |
# Batch size per device for evaluation
|
| 32 |
per_device_eval_batch_size: 1
|
| 33 |
# The number of gradient accumulation steps
|
|
|
|
| 27 |
# Seed for random number generator
|
| 28 |
seed: 42
|
| 29 |
# Batch size per device for training
|
| 30 |
+
per_device_train_batch_size: 4
|
| 31 |
# Batch size per device for evaluation
|
| 32 |
per_device_eval_batch_size: 1
|
| 33 |
# The number of gradient accumulation steps
|