SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

distillation: fix ModuleNotFoundError error in token counts script

Browse Source
This commit is contained in:
Stefan Schweter
2019-08-31 12:21:38 +02:00
parent d7a4c3252e
commit a1c34bd286
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/distillation/scripts/token_counts.py
View File

@@ -19,7 +19,7 @@ from collections import Counter
import argparse import argparse
import pickle import pickle
from utils import logger from examples.distillation.utils import logger
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser(description="Token Counts for smoothing the masking probabilities in MLM (cf XLM/word2vec)") parser = argparse.ArgumentParser(description="Token Counts for smoothing the masking probabilities in MLM (cf XLM/word2vec)")