Fix model saving for older versions of PyTorch, see #71

GAMES-UChile · Dec 20, 2023 · 2c3b533 · 2c3b533
1 parent d36d227
commit 2c3b533
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mogptk/gpr/model.py b/mogptk/gpr/model.py
@@ -129,7 +129,8 @@ def compile(self):
             self._compiled_forward = torch.jit.trace(self.forward, ())
 
     def __getstate__(self):
-        state = super().__getstate__()
+        state = self.__dict__.copy()
+        state.pop('_compiled_call_impl', None)
         state['_modules'] = state['_modules'].copy()
         state['_modules'].pop('_compiled_forward', None)
         return state