Update

Author: svekars

conf.py

@@ -107,13 +107,6 @@ def reset_seeds(gallery_conf, fname):
     random.seed(10)
     numpy.random.seed(10)
     torch.set_grad_enabled(True)
-    
-    # Reset any patching from memory_format_tutorial
-    if hasattr(torch, "old_attrs"):
-        for (m, attrs) in torch.old_attrs.items():
-            for (k, v) in attrs.items():
-                setattr(m, k, v)
-        delattr(torch, "old_attrs")
 
 
     gc.collect()

intermediate_source/memory_format_tutorial.py

@@ -376,6 +376,20 @@ def attribute(m):
     for (k, v) in attrs.items():
         setattr(m, k, v)
 
+# Recover Tensor
+torch.Tensor = old_attrs[torch.Tensor]["Tensor"]
+# Recover nn.functional
+torch.nn.functional = old_attrs[torch.nn.functional]["nn.functional"]
+# Recover torch
+torch.is_cuda = old_attrs[torch]["is_cuda"]
+torch.has_names = old_attrs[torch]["has_names"]
+torch.numel = old_attrs[torch]["numel"]
+torch.stride = old_attrs[torch]["stride"]
+torch.is_contiguous = old_attrs[torch]["is_contiguous"]
+torch.__class__ = old_attrs[torch]["__class__"]
+del old_attrs
+
+
 ######################################################################
 # Work to do
 # ----------