Don't use dill for pickle

inferno-pytorch · Aug 7, 2018 · 6792b97 · 6792b97
1 parent bfa1867
commit 6792b97
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/inferno/trainers/basic.py b/inferno/trainers/basic.py
@@ -1593,8 +1593,8 @@ def save(self, exclude_loader=True, stash_best_checkpoint=True):
 
         # Save the state dictionary
         torch.save(self.get_config(exclude_loader=exclude_loader),
-                   checkpoint_path,
-                   pickle_module=dill)
+                   checkpoint_path)
+                   # pickle_module=dill)
 
         self.callbacks.call(self.callbacks.END_OF_SAVE,
                             save_to_directory=self._save_to_directory,