Home Explore Help
Sign In
brust
/
chillax
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

Added missing maybe_update_embedding to restore

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

+ 1 - 0
chillax/chillax_classifier.py
View File 

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

+ 1 - 1
chillax/version.py
View File 

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