Fixing --release option

urialon · urialon · commit 29a4c46454fc · 2019-12-05T08:48:54.000+02:00
diff --git a/code2vec.py b/code2vec.py
@@ -27,7 +27,7 @@ def load_model_dynamically(config: Config) -> Code2VecModelBase:
     if config.SAVE_T2V is not None:
         model.save_word2vec_format(config.SAVE_T2V, VocabType.Target)
         config.log('Target word vectors saved in word2vec text format in: %s' % config.SAVE_T2V)
-    if config.is_testing and not config.is_training:
+    if (config.is_testing and not config.is_training) or config.RELEASE:
         eval_results = model.evaluate()
         if eval_results is not None:
             config.log(
diff --git a/model_base.py b/model_base.py
@@ -41,7 +41,8 @@ def __init__(self, config: Config):
 
         self._log_creating_model()
 
-        self._init_num_of_examples()
+        if not config.RELEASE:
+            self._init_num_of_examples()
         self._log_model_configuration()
         self.vocabs = Code2VecVocabs(config)
         self.vocabs.target_vocab.get_index_to_word_lookup_table()  # just to initialize it (if not already initialized)
@@ -53,7 +54,7 @@ def _log_creating_model(self):
         self.log('')
         self.log('---------------------------------------------------------------------')
         self.log('---------------------------------------------------------------------')
-        self.log('---------------------- Creating word2vec model ----------------------')
+        self.log('---------------------- Creating code2vec model ----------------------')
         self.log('---------------------------------------------------------------------')
         self.log('---------------------------------------------------------------------')
 
