fix from_config

ai-forever · Jan 19, 2019 · 731d39b · 731d39b
1 parent 040c9f2
commit 731d39b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/train/train.py b/modules/train/train.py
@@ -195,10 +195,10 @@ def save_config(self, path):
             json.dump(self.config, file)
 
     @classmethod
-    def from_config(cls, path):
+    def from_config(cls, path, for_train=True):
         with open(path, "r") as file:
             config = json.load(file)
-        data = BertNerData.from_config(config["data"])
+        data = BertNerData.from_config(config["data"], for_train)
         name = config["model"]["name"]
         # TODO: release all models (now only for BertBiLSTMNCRF)
         if name not in released_models: