Update

Author: svekars

.jenkins/validate_tutorials_built.py

@@ -54,7 +54,6 @@
     "advanced_source/coding_ddpg", # reenable after 3302 is fixed
     "intermediate_source/torchrec_intro_tutorial", # reenable after 3302 is fixed
     "recipes_source/recipes/reasoning_about_shapes", # reenable after 3326 is fixed
-    "intermediate_source/memory_format_tutorial"
 ]
 
 def tutorial_source_dirs() -> List[Path]:

conf.py

@@ -121,9 +121,13 @@ def reset_seeds(gallery_conf, fname):
     torch.set_grad_enabled(True)
     torch.set_default_dtype(torch.float32)
     torch.set_default_tensor_type(torch.FloatTensor)
-    import logging
+    
+    # Reset torch modules that might have been modified by tutorials
+    import importlib
+    importlib.reload(torch)
+    if 'torchvision' in sys.modules:
+        importlib.reload(torchvision)
 
-    logging.getLogger().setLevel(logging.WARNING)
     gc.collect()