Here is my code from transformers import Seq2SeqTrainer,Seq2SeqTrainingArguments, EarlyStoppingCallback, BertTokenizer,MT5ForConditionalGeneration from transformers.data.data_collator import DataCollatorForSeq2Seq,default_data_collator from torch.utils.data import Dataset, DataLoader import pandas …

Not sure how to compute BLEU through compute_metrics

xiami May 30, 2023, 6:25am 5

set predict_with_generate = True in training_args solved my problem

1 Like

Problem with custom metric for custom T5 model