Fix serialized models

soulitzer · soulitzer · commit b3d6c9cd54e7 · 2021-08-17T17:53:46.000-04:00
diff --git a/torch/autograd/function.py b/torch/autograd/function.py
@@ -171,6 +171,9 @@ def set_materialize_grads(self, value: bool):
         """
         self.materialize_grads = value
 
+# DO NOT USE: This is only defined to be able to load old serialized models
+_ContextMethodMixin = FunctionCtx
+
 class _HookMixin(object):
 
     @staticmethod
