Kevin Fink
commited on
Commit
·
229da9e
1
Parent(s):
37472a5
init
Browse files
app.py
CHANGED
|
@@ -28,7 +28,6 @@ def fine_tune_model(model_name, dataset_name, hub_id, num_epochs, batch_size, lr
|
|
| 28 |
per_device_eval_batch_size=batch_size,
|
| 29 |
num_train_epochs=num_epochs,
|
| 30 |
weight_decay=0.01,
|
| 31 |
-
evaluation_strategy='epoch',
|
| 32 |
gradient_accumulation_steps=grad,
|
| 33 |
load_best_model_at_end=True,
|
| 34 |
metric_for_best_model="accuracy",
|
|
|
|
| 28 |
per_device_eval_batch_size=batch_size,
|
| 29 |
num_train_epochs=num_epochs,
|
| 30 |
weight_decay=0.01,
|
|
|
|
| 31 |
gradient_accumulation_steps=grad,
|
| 32 |
load_best_model_at_end=True,
|
| 33 |
metric_for_best_model="accuracy",
|