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

make style

Browse Source
This commit is contained in:
Patrick von Platen
2020-03-13 16:35:52 +01:00
parent c2ee3840ae
commit 4f75d380a4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/summarization/bart/evaluate_cnn.py
View File

@@ -36,7 +36,7 @@ def generate_summaries(lns, out_file, batch_size=8, device=DEFAULT_DEVICE):
no_repeat_ngram_size=3, no_repeat_ngram_size=3,
early_stopping=True, early_stopping=True,
do_sample=False, do_sample=False,
decoder_start_token_id=model.config.eos_token_ids[0] decoder_start_token_id=model.config.eos_token_ids[0],
) )
dec = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in summaries] dec = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in summaries]
for hypothesis in dec: for hypothesis in dec: