From dc117755436fe436161491ab0e16ce3d21bbd275 Mon Sep 17 00:00:00 2001 From: StevenTang1998 <1020139164@qq.com> Date: Thu, 15 Apr 2021 21:21:03 +0800 Subject: [PATCH] FIX: bugs in readme --- README.md | 12 ++++++------ README_CN.md | 2 +- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/README.md b/README.md index 58c88c37..c913442d 100644 --- a/README.md +++ b/README.md @@ -130,7 +130,7 @@ python run_textbox.py --model=GPT2 --dataset=COCO \ --pretrained_model_path=pretrained_model/gpt2 ``` -### **Train with Distributed Data Parallel** +### **Train with Distributed Data Parallel (DDP)** TextBox supports to train models with multiple GPUs conveniently. You don't need to modify the model, just run the following command: @@ -418,7 +418,7 @@ NLL, BLEU and SBLEU on test dataset: | **MaskGAN** | 509.58 | 56.61 | 21.41 | 4.49 | 0.86 | 92.09 | 77.88 | 59.62 | 42.36 | | **GPT-2** | 348.67 | 72.52 | 41.75 | 15.40 | 4.22 | 86.21 | 58.26 | 30.03 | 12.56 | -Part of generated examples (with `max_length` 100): +Part of generated examples (with `max_length=100`): @@ -445,7 +445,7 @@ Part of generated examples (with `max_length` 100): #### GigaWord (Summarization) -ROUGE metric on test dataset using beam search (with `beam_size` 5): +ROUGE metric on test dataset using beam search (with `beam_size=5`):
@@ -534,7 +534,7 @@ Part of generated examples: #### IWSLT2014 German-English (Translation) -BLEU metric on test dataset with three decoding strategies: top-k sampling, greedy search and beam search (with `beam_size` 5): +BLEU metric on test dataset with three decoding strategies: top-k sampling, greedy search and beam search (with `beam_size=5`):
@@ -630,7 +630,7 @@ Part of generated examples: #### Persona Chat (Dialogue) -BLEU and distinct metrics on test dataset using beam search (with `beam_size` 5): +BLEU and distinct metrics on test dataset using beam search (with `beam_size=5`):
@@ -675,7 +675,7 @@ BLEU and distinct metrics on test dataset using beam search (with `beam_size` 5) #### Amazon Electronic (Attribute to text) -BLEU and distinct metrics on test dataset using beam search (with `beam_size` 5): +BLEU and distinct metrics on test dataset using beam search (with `beam_size=5`):
diff --git a/README_CN.md b/README_CN.md index dded0101..f71d34ef 100644 --- a/README_CN.md +++ b/README_CN.md @@ -417,7 +417,7 @@ python -m torch.distributed.launch --nproc_per_node=[gpu_num] \ | **MaskGAN** | 509.58 | 56.61 | 21.41 | 4.49 | 0.86 | 92.09 | 77.88 | 59.62 | 42.36 | | **GPT-2** | 348.67 | 72.52 | 41.75 | 15.40 | 4.22 | 86.21 | 58.26 | 30.03 | 12.56 | -部分生成实例展示(最大长度 `max_length` 为100): +部分生成实例展示(最大长度 `max_length`设置为100 ):