Update

Author: svekars

intermediate_source/memory_format_tutorial.py

@@ -379,14 +379,17 @@ def attribute(m):
 import gc
 import sys
 
-if torch.nn.functional in old_attrs:
-    for k, v in old_attrs[torch.nn.functional].items():
-        setattr(torch.nn.functional, k, v)
-
 torch._dynamo.reset()
 if torch.cuda.is_available():
     torch.cuda.empty_cache()
 
+gc.collect()
+
+if hasattr(torch, "_check_wrapper_patched"):
+    delattr(torch, "_check_wrapper_patched")
+
+old_attrs.clear()
+
 ######################################################################
 # Work to do
 # ----------