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