Merge pull request #962 from rizar/change_pickle_protocol

Use compression for saving separately
mila-iqia · Jan 28, 2016 · 7510326 · 7510326
2 parents fa3a290 + 249cae2
commit 7510326
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/blocks/extensions/saveload.py b/blocks/extensions/saveload.py
@@ -97,7 +97,8 @@ def do(self, callback_name, *args):
             filenames = self.save_separately_filenames(path)
             for attribute in self.save_separately:
                 secure_dump(getattr(self.main_loop, attribute),
-                            filenames[attribute], cPickle.dump)
+                            filenames[attribute], cPickle.dump,
+                            protocol=cPickle.HIGHEST_PROTOCOL)
         except Exception:
             path = None
             raise