HF_Final_Assignment_Template

Sleeping

Samuel Thomas commited on May 31

Commit

9fd7617

1 Parent(s): 8792b9a

remove cuda

Files changed (1) hide show

tools.py CHANGED Viewed

@@ -3660,8 +3660,8 @@ Current date: {current_date_str}
     # Cleanup
     if result.get("done"):
-        torch.cuda.empty_cache()
-        torch.cuda.ipc_collect()
         gc.collect()
         print("🧹 Released GPU memory after completion")

     # Cleanup
     if result.get("done"):
+        #torch.cuda.empty_cache()
+        #torch.cuda.ipc_collect()
         gc.collect()
         print("🧹 Released GPU memory after completion")