[s2strainer] fix eval dataset loading (#7477)

This commit is contained in:
Suraj Patil
2020-09-30 22:09:13 +05:30
committed by GitHub
parent a97a73e0ee
commit 08939cfdf7

View File

@@ -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 = (