SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[s2strainer] fix eval dataset loading (#7477)

Browse Source
This commit is contained in:
Suraj Patil
2020-09-30 22:09:13 +05:30
committed by GitHub
parent a97a73e0ee
commit 08939cfdf7
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/seq2seq/finetune_trainer.py
View File

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