diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index 0f22bdb6..c7cd6395 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -187,7 +187,7 @@ class DevicePoolExecutor: def submit(self, key: str, fn: Callable[..., None], /, *args, needs_device: Optional[DeviceParams] = None, **kwargs) -> None: device = self.get_next_device(needs_device=needs_device) - logger.info("assigning job %s to device %s", key, device) + logger.info("assigning job %s to device %s: %s", key, device, self.devices[device]) context = JobContext(key, self.devices, device_index=device) future = self.pool.submit(fn, context, *args, **kwargs)