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

[s2strainer] fix eval dataset loading (#7477)

parent a97a73e0
......@@ -22,6 +22,7 @@ from transformers import (
set_seed,
)
from transformers.modeling_bart import shift_tokens_right
from transformers.trainer_utils import EvaluationStrategy
from utils import (
LegacySeq2SeqDataset,
Seq2SeqDataset,
......@@ -350,7 +351,7 @@ def main():
max_source_length=data_args.max_source_length,
prefix=model.config.prefix or "",
)
if training_args.do_eval
if training_args.do_eval or training_args.evaluation_strategy != EvaluationStrategy.NO
else None
)
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