Rolv-Arild commited on
Commit
e58b850
·
1 Parent(s): 57f88c8

Fix KeyError

Browse files
Files changed (1) hide show
  1. run_speech_recognition_ctc.py +1 -1
run_speech_recognition_ctc.py CHANGED
@@ -738,7 +738,7 @@ def main():
738
  args=training_args,
739
  compute_metrics=compute_metrics,
740
  train_dataset=vectorized_datasets["train"] if training_args.do_train else None,
741
- eval_dataset=vectorized_datasets["eval"] if training_args.do_eval else None,
742
  tokenizer=feature_extractor,
743
  )
744
 
 
738
  args=training_args,
739
  compute_metrics=compute_metrics,
740
  train_dataset=vectorized_datasets["train"] if training_args.do_train else None,
741
+ eval_dataset=vectorized_datasets["validation"] if training_args.do_eval else None,
742
  tokenizer=feature_extractor,
743
  )
744