Skip to content

Commit

Permalink
Closing bracket in DistilBERT's token count.
Browse files Browse the repository at this point in the history
  • Loading branch information
LysandreJik committed Aug 29, 2019
1 parent fe8fb10 commit caf1d11
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/distillation/scripts/token_counts.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
if __name__ == '__main__':
parser = argparse.ArgumentParser(description="Token Counts for smoothing the masking probabilities in MLM (cf XLM/word2vec)")
parser.add_argument("--data_file", type=str, default="data/dump.bert-base-uncased.pickle",
help="The binarized dataset."
help="The binarized dataset.")
parser.add_argument("--token_counts_dump", type=str, default="data/token_counts.bert-base-uncased.pickle",
help="The dump file.")
parser.add_argument("--vocab_size", default=30522, type=int)
Expand Down

0 comments on commit caf1d11

Please sign in to comment.