update README.md

neulab · Mar 1, 2022 · c0dcc5f · c0dcc5f
1 parent a263b6a
commit c0dcc5f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/train/README.md b/train/README.md
@@ -18,7 +18,7 @@ python bart.py --train_file train.json --validation_file val.json --output_dir m
 Then you can use your custom BARTScore for evaluation.
 
 ## Reproduce
-To reproduce our results, please see the folder [`reproduce`](reproduce). Due to limited computing resources, we sharded the BART into multiple GPUs and trained the model. See [`reproduce/finetune.py`](reproduce/finetune.py) for details.
+To reproduce our results, please see the folder [`reproduce`](reproduce). Due to limited computing resources, we sharded the BART into multiple GPUs and trained the model, please see [`reproduce/finetune.py`](reproduce/finetune.py) for details.
 
 
 
diff --git a/train/reproduce/data/README.md b/train/reproduce/data/README.md
@@ -0,0 +1,7 @@
+# Data
+
+The Parabank2 dataset can be found here: [https://nlp.jhu.edu/parabank/](https://nlp.jhu.edu/parabank/). Please reformat the data into a `parabank2.json` file like this if you want to use our provided script:
+```
+{"text": "1994 German Grand Prix", "summary": "1994 Grand Prix of Germany"}
+{"text": "2004 OFC Nations Cup", "summary": "2004 Ocean Cup of Nations"}
+```