Kevin Fink commited on
Commit
648a87e
·
1 Parent(s): b755227
Files changed (1) hide show
  1. app.py +1 -1
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
  )