apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

log device options

This commit is contained in:
Sean Sube 2023-02-04 21:26:11 -06:00
parent cc12cb0fcf
commit 9d1f9412f6
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/params.py
View File

@ -66,7 +66,7 @@ class DeviceParams:
self.options = options
def __str__(self) -> str:
return '%s - %s' % (self.device, self.provider)
return '%s - %s (%s)' % (self.device, self.provider, self.options)
def torch_device(self) -> str:
if self.device.startswith('cuda'):