From c77621292bf130779d1572f87155ffe1eeb0fe5a Mon Sep 17 00:00:00 2001 From: xuyige Date: Wed, 29 Aug 2018 15:26:18 +0800 Subject: [PATCH] fix a bug in char_language_model.py --- fastNLP/models/char_language_model.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/fastNLP/models/char_language_model.py b/fastNLP/models/char_language_model.py index a9f40ae9..f678070e 100644 --- a/fastNLP/models/char_language_model.py +++ b/fastNLP/models/char_language_model.py @@ -142,6 +142,8 @@ class CharLM(BaseModel): "char_dict": char_dict, "reverse_word_dict": reverse_word_dict, } + if not os.path.exists("cache"): + os.mkdir("cache") torch.save(objects, "cache/prep.pt") print("Preprocess done.")