wangjin2000 commited on
Commit
c7e53bf
·
verified ·
1 Parent(s): 2aa5124

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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='loss',
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