Kevin Fink
commited on
Commit
·
8bfc42f
1
Parent(s):
df08c41
dev
Browse files
app.py
CHANGED
|
@@ -127,7 +127,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
|
|
| 127 |
args=training_args,
|
| 128 |
train_dataset=train_dataset,
|
| 129 |
eval_dataset=saved_test_dataset,
|
| 130 |
-
|
| 131 |
)
|
| 132 |
|
| 133 |
elif os.access(f'/data/{hub_id.strip()}_train_dataset3', os.R_OK):
|
|
|
|
| 127 |
args=training_args,
|
| 128 |
train_dataset=train_dataset,
|
| 129 |
eval_dataset=saved_test_dataset,
|
| 130 |
+
compute_metrics=compute_metrics,
|
| 131 |
)
|
| 132 |
|
| 133 |
elif os.access(f'/data/{hub_id.strip()}_train_dataset3', os.R_OK):
|