From a1c34bd286d57625b844703705a39e40737a21ba Mon Sep 17 00:00:00 2001 From: Stefan Schweter Date: Sat, 31 Aug 2019 12:21:38 +0200 Subject: [PATCH] distillation: fix ModuleNotFoundError error in token counts script --- examples/distillation/scripts/token_counts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/distillation/scripts/token_counts.py b/examples/distillation/scripts/token_counts.py index d6b6126fb6..d791c66be3 100644 --- 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)")