Browse Source

Added missing maybe_update_embedding to restore

Clemens-Alexander Brust 4 years ago
parent
commit
f2ab4261df
2 changed files with 2 additions and 1 deletions
  1. 1 0
      chillax/chillax_classifier.py
  2. 1 1
      chillax/version.py

+ 1 - 0
chillax/chillax_classifier.py

@@ -379,6 +379,7 @@ class CHILLAXKerasHC(
             pickle.dump((self.uid_to_dimension,), target)
 
     def restore(self, path):
+        self.maybe_update_embedding()
         with open(path + "_hc.pkl", "rb") as target:
             new_weights = pickle.load(target)
             has_weights = False

+ 1 - 1
chillax/version.py

@@ -1 +1 @@
-__version__ = "0.1a11"
+__version__ = "0.1a12"