Skip to content

Commit c4d8a86

Browse files
Merge pull request #3 from jorge-martinez-gil/jorge-martinez-gil/fix-eval-dataset-trainer
fix: pass train_dataset and eval_dataset to CloneDetectionTrainer constructor
2 parents c8eb885 + b561bd8 commit c4d8a86

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

scripts/run_clone_experiment.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,8 @@ def main() -> None:
114114
trainer = CloneDetectionTrainer(
115115
model=model,
116116
args=get_training_args(args.output_dir, **training_overrides),
117+
train_dataset=train_ds,
118+
eval_dataset=val_ds,
117119
data_collator=DataCollatorWithPadding(
118120
tokenizer=tokenizer,
119121
pad_to_multiple_of=8,

0 commit comments

Comments
 (0)