From 3dca9583bf38a7dbf26df9d9efe64fac49c41456 Mon Sep 17 00:00:00 2001 From: tissue-box Date: Wed, 23 Mar 2022 19:22:34 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dgener=20ende=E6=95=B0?= =?UTF-8?q?=E6=8D=AE=E7=BC=BA=E5=A4=B1=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_de.sh | 2 +- .../built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_ro.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_de.sh b/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_de.sh index 1a202ccee4..7e13b08739 100644 --- a/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_de.sh +++ b/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_de.sh @@ -8,7 +8,7 @@ HYP=hyp REF=ref fairseq-generate $DATA_PATH \ - --fp16 --path $model_dir --max-tokens 4096 \ + --fp16 --path $model_dir --max-tokens 1024 \ --task translation_from_pretrained_bart \ --gen-subset test \ -t de_DE -s en_XX \ diff --git a/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_ro.sh b/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_ro.sh index 66ea0d5ece..18e527b108 100644 --- a/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_ro.sh +++ b/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/generate_on_en_ro.sh @@ -16,7 +16,7 @@ HYP=hyp REF=ref fairseq-generate $DATA_PATH \ - --fp16 --path $model_dir --max-tokens 4096 \ + --fp16 --path $model_dir --max-tokens 1024 \ --task translation_from_pretrained_bart \ --gen-subset test \ -t ro_RO -s en_XX \ -- Gitee