Skip to content

Commit

Permalink
Update eval_retriever_eli5.py
Browse files Browse the repository at this point in the history
  • Loading branch information
martiansideofthemoon authored Aug 14, 2021
1 parent eee25eb commit 82acc90
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion eval_retriever_eli5.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
import tensorflow_hub as hub
from transformers import AutoTokenizer

VOCAB_PATH = "/mnt/nfs/work1/miyyer/simengsun/in-book-retrieval/RT-data/vocab.pg19_length8k.32768.subwords"
VOCAB_PATH = "vocab.pg19_length8k.32768.subwords"

parser = argparse.ArgumentParser()
parser.add_argument('--retrieval_corpus', default="eli5_train", type=str, choices=["eli5_train", "kilt_wiki"])
Expand Down

0 comments on commit 82acc90

Please sign in to comment.