Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -88,7 +88,7 @@ def Finetune(base_model_path): #, train_dataset, test_dataset):
|
|
| 88 |
evaluation_strategy="epoch",
|
| 89 |
load_best_model_at_end=True,
|
| 90 |
save_strategy='epoch',
|
| 91 |
-
metric_for_best_model='
|
| 92 |
save_total_limit = 5,
|
| 93 |
gradient_accumulation_steps=2,
|
| 94 |
push_to_hub=True, #jw 20240918
|
|
|
|
| 88 |
evaluation_strategy="epoch",
|
| 89 |
load_best_model_at_end=True,
|
| 90 |
save_strategy='epoch',
|
| 91 |
+
metric_for_best_model='eval_loss',
|
| 92 |
save_total_limit = 5,
|
| 93 |
gradient_accumulation_steps=2,
|
| 94 |
push_to_hub=True, #jw 20240918
|