fix dataset key

GRAAL-Research · Nov 14, 2023 · 1adf606 · 1adf606
1 parent 9c8d7f6
commit 1adf606
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/training/few_shot_training.py b/src/training/few_shot_training.py
@@ -98,7 +98,7 @@ def tokenize_function(example):
     model,
     training_args,
     train_dataset=tokenized_csmd_dataset["train"],
-    eval_dataset=tokenized_csmd_dataset["val"],
+    eval_dataset=tokenized_csmd_dataset["dev"],
     data_collator=data_collator,
     tokenizer=tokenizer,
     compute_metrics=compute_metrics,