Kevin Fink
commited on
Commit
·
648a87e
1
Parent(s):
b755227
deve
Browse files
app.py
CHANGED
@@ -235,7 +235,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
|
|
235 |
args=training_args,
|
236 |
train_dataset=train_set['train'],
|
237 |
eval_dataset=train_set['validation'],
|
238 |
-
compute_metrics=compute_metrics,
|
239 |
#data_collator=data_collator,
|
240 |
#processing_class=tokenizer,
|
241 |
)
|
|
|
235 |
args=training_args,
|
236 |
train_dataset=train_set['train'],
|
237 |
eval_dataset=train_set['validation'],
|
238 |
+
#compute_metrics=compute_metrics,
|
239 |
#data_collator=data_collator,
|
240 |
#processing_class=tokenizer,
|
241 |
)
|