Skip to content

Commit 1bdd7f8

Browse files
Update instructions for pre-trained model
1 parent b83c7a9 commit 1bdd7f8

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

README.md

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -208,9 +208,10 @@ It also seems to overfit as sometimes it will just pop out sentences from its tr
208208

209209
## Pretrained model
210210

211-
You can find a pre-trained model [here](https://drive.google.com/file/d/0Bw-phsNSkq23TXltWGlOdk9wOXc/view?usp=sharing), trained of the default corpus. To launch it, extract it inside `DeepQA/save/` and run `./main.py --modelTag pretrainedv2 --test interactive`. The old pre-trained model is still available [here](https://drive.google.com/file/d/0Bw-phsNSkq23amlSZXVqcm5oVFU/view?usp=sharing) (Won't work with the current version).
212-
213-
**Warning**: The current pre-trained model has been created with a previous version of the program where the dataset format was slightly different. In order to make it works, you have to use the old dataset file by copying the `.pkl` file present in the downloaded folder into `data/samples/`.
211+
You can find a pre-trained model [here](https://drive.google.com/file/d/0Bw-phsNSkq23TXltWGlOdk9wOXc/view?usp=sharing), trained of the default corpus. To use it:
212+
1. Extract the zip file inside `DeepQA/save/`
213+
2. Copy the preprocessed dataset from `save/model-pretrainedv2/dataset-cornell-old-lenght10-filter0.pkl` to `data/samples/`.
214+
3. Run `./main.py --modelTag pretrainedv2 --test interactive`.
214215

215216
If you have a high-end GPU, don't hesitate to play with the hyper-parameters/corpus to train a better model. From my experiments, it seems that the learning rate and dropout rate have the most impact on the results. Also if you want to share your models, don't hesitate to contact me and I'll add it here.
216217

0 commit comments

Comments
 (0)