From 4e5ad54471115580badc1c220b17dc0b80f0c7bd Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 4 Feb 2023 14:01:30 -0600 Subject: [PATCH] improve device logging --- api/onnx_web/device_pool.py | 4 ++-- api/onnx_web/serve.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index ecde3877..936065eb 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -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: diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index 249c4ca4..2048f54f 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -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()