Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -42,8 +42,8 @@ training_args = TrainingArguments(
|
|
| 42 |
output_dir="./results",
|
| 43 |
evaluation_strategy="epoch",
|
| 44 |
learning_rate=2e-5,
|
| 45 |
-
per_device_train_batch_size=
|
| 46 |
-
per_device_eval_batch_size=
|
| 47 |
num_train_epochs=3,
|
| 48 |
weight_decay=0.01,
|
| 49 |
save_total_limit=2,
|
|
|
|
| 42 |
output_dir="./results",
|
| 43 |
evaluation_strategy="epoch",
|
| 44 |
learning_rate=2e-5,
|
| 45 |
+
per_device_train_batch_size=2,
|
| 46 |
+
per_device_eval_batch_size=2,
|
| 47 |
num_train_epochs=3,
|
| 48 |
weight_decay=0.01,
|
| 49 |
save_total_limit=2,
|