Kevin Fink commited on
Commit
aeb71da
·
1 Parent(s): d37e5b1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -134,7 +134,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
134
  model=model,
135
  args=training_args,
136
  train_dataset=train_dataset,
137
- eval_dataset=dataset['validation'],
138
  compute_metrics=compute_metrics,
139
  data_collator=data_collator,
140
  #processing_class=tokenizer,
 
134
  model=model,
135
  args=training_args,
136
  train_dataset=train_dataset,
137
+ eval_dataset=saved_test_dataset['input_ids'],
138
  compute_metrics=compute_metrics,
139
  data_collator=data_collator,
140
  #processing_class=tokenizer,