empty cuda cache

clee2000 · clee2000 · commit ef487aeadf90 · 2023-06-27T10:38:01.000-07:00
diff --git a/conf.py b/conf.py
@@ -91,6 +91,7 @@
 def reset_seeds(gallery_conf, fname):
     torch.manual_seed(42)
     torch.set_default_device(None)
+    torch.cuda.empty_cache()
     random.seed(10)
     numpy.random.seed(10)
     gc.collect()
