tc

Author: clee2000

conf.py

@@ -98,8 +98,6 @@
 # -- Sphinx-gallery configuration --------------------------------------------
 
 def reset_seeds(gallery_conf, fname):
-    import importlib
-    importlib.reload(torch)
     torch.cuda.empty_cache()
     torch.backends.cudnn.deterministic = True
     torch.backends.cudnn.benchmark = False
@@ -128,6 +126,7 @@ def reset_seeds(gallery_conf, fname):
     'pypandoc': {'extra_args': ['--mathjax', '--toc'],
                  'filters': ['.jenkins/custom_pandoc_filter.py'],
     },
+    'parallel': True,
 }
 
 html_baseurl = 'https://pytorch.org/tutorials/' # needed for sphinx-sitemap