Эх сурвалжийг харах

Added missing maybe_update_embedding to restore

Clemens-Alexander Brust 4 жил өмнө
parent
commit
f2ab4261df

+ 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"