1 batch size
Browse files- scripts/train.py +1 -1
scripts/train.py
CHANGED
|
@@ -72,7 +72,7 @@ def main():
|
|
| 72 |
# Training arguments
|
| 73 |
training_args = TrainingArguments(
|
| 74 |
output_dir="./output",
|
| 75 |
-
per_device_train_batch_size=
|
| 76 |
gradient_accumulation_steps=8,
|
| 77 |
learning_rate=1e-5,
|
| 78 |
num_train_epochs=1,
|
|
|
|
| 72 |
# Training arguments
|
| 73 |
training_args = TrainingArguments(
|
| 74 |
output_dir="./output",
|
| 75 |
+
per_device_train_batch_size=1,
|
| 76 |
gradient_accumulation_steps=8,
|
| 77 |
learning_rate=1e-5,
|
| 78 |
num_train_epochs=1,
|