Kevin Fink
commited on
Commit
·
b3ffd25
1
Parent(s):
c812c62
init
Browse files
app.py
CHANGED
|
@@ -54,7 +54,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
|
|
| 54 |
per_device_eval_batch_size=1,
|
| 55 |
num_train_epochs=int(num_epochs),
|
| 56 |
weight_decay=0.01,
|
| 57 |
-
gradient_accumulation_steps=grad*0.1,
|
| 58 |
load_best_model_at_end=True,
|
| 59 |
metric_for_best_model="accuracy",
|
| 60 |
greater_is_better=True,
|
|
|
|
| 54 |
per_device_eval_batch_size=1,
|
| 55 |
num_train_epochs=int(num_epochs),
|
| 56 |
weight_decay=0.01,
|
| 57 |
+
#gradient_accumulation_steps=grad*0.1,
|
| 58 |
load_best_model_at_end=True,
|
| 59 |
metric_for_best_model="accuracy",
|
| 60 |
greater_is_better=True,
|