fix(api): log number of active threads when running GC (#170)

ssube · Feb 18, 2023 · d09446c · d09446c
1 parent aec44aa
commit d09446c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/api/onnx_web/utils.py b/api/onnx_web/utils.py
@@ -1,6 +1,7 @@
 import gc
 from logging import getLogger
 from os import environ, path
+import threading
 from typing import Any, Dict, List, Optional, Union
 
 import torch
@@ -135,7 +136,7 @@ def get_size(val: Union[int, str, None]) -> SizeChart:
 
 
 def run_gc(devices: List[DeviceParams] = []):
-    logger.debug("running garbage collection")
+    logger.debug("running garbage collection with %s active threads", threading.active_count())
     gc.collect()
 
     if torch.cuda.is_available():