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 ):