apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

improve device logging

This commit is contained in:
Sean Sube 2023-02-04 14:01:30 -06:00
parent 2c9a10e2b7
commit 4e5ad54471
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/device_pool.py
View File

@ -101,11 +101,11 @@ class DevicePoolExecutor:
device_count = len(devices)
if pool is None:
logger.info(
'creating thread pool executor for %s devices: %s', device_count, devices)
'creating thread pool executor for %s devices: %s', device_count, [d.device for d in devices])
self.pool = ThreadPoolExecutor(device_count)
else:
logger.info('using existing pool for %s devices: %s',
device_count, devices)
device_count, [d.device for d in devices])
self.pool = pool
def cancel(self, key: str) -> bool:

2
api/onnx_web/serve.py
View File

@ -322,7 +322,7 @@ def load_platforms():
else:
available_platforms.append(DeviceParams(potential, platform_providers[potential]))
logger.info('available acceleration platforms: %s', available_platforms)
logger.info('available acceleration platforms: %s', ', '.join([str(p) for p in available_platforms]))
context = ServerContext.from_environ()