Add torch.cuda.empty_cache() to the reset_seeds function (#2492)

Svetlana Karslioglu · web-flow · commit 081f7df7a860 · 2023-06-27T12:30:35.000-07:00
Adding torch.cuda.empty_cache() to the reset_seeds function to solve CUDA out of memory error.
diff --git a/conf.py b/conf.py
@@ -87,6 +87,7 @@
 # -- Sphinx-gallery configuration --------------------------------------------
 
 def reset_seeds(gallery_conf, fname):
+    torch.cuda.empty_cache()
     torch.manual_seed(42)
     torch.set_default_device(None)
     random.seed(10)
