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