Add statement regarding the paper version of simple-lstm

EhsanMashhadi · Dec 21, 2021 · 554afff · 554afff
1 parent 68983f6
commit 554afff
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -37,7 +37,9 @@ You can find the paper here: https://arxiv.org/abs/2103.11626
 4. Evaluate the model
     - `bash ./scripts/simple-lstm/test.sh`
 
-#### Running Simple LSTM Experiments using the legacy version of OpenNMT-py
+### Running Simple LSTM Experiments using the legacy version of OpenNMT-py
+**(This is the original version used to run the simple LSTM experiments in the paper.)**
+
 1. Install [OpenNMT-py legacy](https://github.com/OpenNMT/OpenNMT-py/tree/legacy)
     - `pip install OpenNMT-py==1.2.0`
 2. Preprocess the MSR data

diff --git a/scripts/simple-lstm/build_vocab.sh b/scripts/simple-lstm/build_vocab.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 
 size=small  # Can be: small OR large
-type=repetition  # Can be: repetition OR unique
+type=unique  # Can be: repetition OR unique
 data_config=./scripts/simple-lstm/${type}_${size}_data.yaml
 output_dir=./saved_models/simple-lstm/$type/$size