Update conf.py

malfet · web-flow · commit 5f2160cf0f5f · 2023-06-07T09:08:48.000-07:00
diff --git a/conf.py b/conf.py
@@ -88,7 +88,7 @@
 
 def reset_seeds(gallery_conf, fname):
     torch.manual_seed(42)
-    torch.set_default_device('cpu')
+    torch.set_default_device(None)
     random.seed(10)
 
 sphinx_gallery_conf = {
