Spaces:
Sleeping
Sleeping
Update finetune_full.py
Browse files- finetune_full.py +1 -1
finetune_full.py
CHANGED
|
@@ -58,7 +58,7 @@ training_args = TrainingArguments(
|
|
| 58 |
warmup_ratio=0.03,
|
| 59 |
lr_scheduler_type="cosine",
|
| 60 |
report_to="tensorboard",
|
| 61 |
-
|
| 62 |
eval_steps=100,
|
| 63 |
save_total_limit=2,
|
| 64 |
load_best_model_at_end=True,
|
|
|
|
| 58 |
warmup_ratio=0.03,
|
| 59 |
lr_scheduler_type="cosine",
|
| 60 |
report_to="tensorboard",
|
| 61 |
+
eval_strategy="steps",
|
| 62 |
eval_steps=100,
|
| 63 |
save_total_limit=2,
|
| 64 |
load_best_model_at_end=True,
|