Skip to content

GitLab

Unverified Commit 08939cfd authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

[s2strainer] fix eval dataset loading (#7477)

parent a97a73e0
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
examples/seq2seq/finetune_trainer.py
View file @ 08939cfd
...@@ -22,6 +22,7 @@ from transformers import ( ...@@ -22,6 +22,7 @@ from transformers import (
set_seed, set_seed,
) )
from transformers.modeling_bart import shift_tokens_right from transformers.modeling_bart import shift_tokens_right
from transformers.trainer_utils import EvaluationStrategy
from utils import ( from utils import (
LegacySeq2SeqDataset, LegacySeq2SeqDataset,
Seq2SeqDataset, Seq2SeqDataset,
...@@ -350,7 +351,7 @@ def main(): ...@@ -350,7 +351,7 @@ def main():
max_source_length=data_args.max_source_length, max_source_length=data_args.max_source_length,
prefix=model.config.prefix or "", prefix=model.config.prefix or "",
) )
if training_args.do_eval if training_args.do_eval or training_args.evaluation_strategy != EvaluationStrategy.NO
else None else None
) )
test_dataset = ( test_dataset = (
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment