Fix missing attribute when loading model from older version (#42242) (#42290)

sharvil · facebook-github-bot · commit 832b1659e78b · 2020-07-31T09:03:07.000-07:00
Summary: Fixes #42242 Pull Request resolved: #42290 Reviewed By: VitalyFedyunin Differential Revision: D22844096 Pulled By: albanD fbshipit-source-id: 707e552e0ed581fbe00f1527ab7426880edaed64
diff --git a/torch/nn/modules/module.py b/torch/nn/modules/module.py
@@ -758,6 +758,8 @@ def __setstate__(self, state):
             self._state_dict_hooks = OrderedDict()
         if '_load_state_dict_pre_hooks' not in self.__dict__:
             self._load_state_dict_pre_hooks = OrderedDict()
+        if '_non_persistent_buffers_set' not in self.__dict__:
+            self._non_persistent_buffers_set = set()
 
     def __getattr__(self, name: str) -> Union[Tensor, 'Module']:
         if '_parameters' in self.__dict__:
