tc

clee2000 · clee2000 · commit 40aba9fd16fe · 2025-05-06T11:31:25.000-07:00
diff --git a/conf.py b/conf.py
@@ -98,6 +98,8 @@
 # -- 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
