apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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

This commit is contained in:
Sean Sube 2023-02-17 21:56:03 -06:00
parent aec44aaaeb
commit d09446ca68
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/utils.py
View File

@ -1,6 +1,7 @@
import gc import gc
from logging import getLogger from logging import getLogger
from os import environ, path from os import environ, path
import threading
from typing import Any, Dict, List, Optional, Union from typing import Any, Dict, List, Optional, Union
import torch import torch
@ -135,7 +136,7 @@ def get_size(val: Union[int, str, None]) -> SizeChart:
def run_gc(devices: List[DeviceParams] = []): 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() gc.collect()
if torch.cuda.is_available(): if torch.cuda.is_available():