tiny fix

2023-06-07 16:02:07 +08:00 · 2023-06-07 16:02:07 +08:00 · 16c2860d56
parent edafb97733
commit 16c2860d56
1 changed files with 1 additions and 1 deletions
--- a/src/api_demo.py
+++ b/src/api_demo.py
@ -29,7 +29,7 @@ from utils import (


 def torch_gc():
-    if not torch.cuda.is_available():
+    if torch.cuda.is_available():
        num_gpus = torch.cuda.device_count()
        for device_id in range(num_gpus):
            with torch.cuda.device(device_id):