#53 fix - cPickle.HIGHEST_PROTOCOL incompatible with Python3

ottokart · Jul 17, 2020 · 8edec70 · 8edec70
1 parent dc2848c
commit 8edec70
Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/data.py b/data.py
@@ -249,7 +249,7 @@ def create_dev_test_train_split_and_vocabulary(root_path, create_vocabulary, tra
                     embeddings.append(e)
 
             with open("We.pcl", 'wb') as f:
-                cPickle.dump(embeddings, f, cPickle.HIGHEST_PROTOCOL)
+                cPickle.dump(embeddings, f)
         else:
             vocabulary = build_vocabulary(word_counts)
         write_vocabulary(vocabulary, WORD_VOCAB_FILE)

diff --git a/models.py b/models.py
@@ -249,7 +249,7 @@ def save(self, file_path, gsums=None, learning_rate=None, validation_ppl_history
         }
 
         with open(file_path, 'wb') as f:
-            cPickle.dump(state, f, protocol=cPickle.HIGHEST_PROTOCOL)
+            cPickle.dump(state, f)
 
 
 class GRUstage2(GRU):