distillation: fix ModuleNotFoundError error in token counts script

2019-08-31 12:21:38 +02:00
parent d7a4c3252e
commit a1c34bd286
1 changed files with 1 additions and 1 deletions
--- a/examples/distillation/scripts/token_counts.py
+++ b/examples/distillation/scripts/token_counts.py
@@ -19,7 +19,7 @@ from collections import Counter
 import argparse
 import pickle

-from utils import logger
+from examples.distillation.utils import logger

 if __name__ == '__main__':
    parser = argparse.ArgumentParser(description="Token Counts for smoothing the masking probabilities in MLM (cf XLM/word2vec)")