Update README.md

This commit is contained in:
Bryan McCann 2018-08-30 11:52:00 -07:00 committed by GitHub
parent 956ea52a5b
commit f4d2e91d93
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -124,7 +124,7 @@ Using a pretrained model or a model you have trained yourself, you can run on ne
```bash
mkdir .data/my_custom_dataset/
touch .datda/my_custom_dataset/val.jsonl
touch .data/my_custom_dataset/val.jsonl
#TODO add examples line by line to val.jsonl in the form of a JSON dict: {"context": "The answer is answer.", "question": "What is the answer?", "answer": "answer"}
nvidia-docker run -it --rm -v `pwd`:/decaNLP/ decanlp bash -c "python /decaNLP/predict.py --evaluate valid --path /decaNLP/mqan_decanlp_qa_first --checkpoint_name model.pth --gpu 0 --tasks my_custom_dataset"
```