log device options
This commit is contained in:
parent
cc12cb0fcf
commit
9d1f9412f6
|
@ -66,7 +66,7 @@ class DeviceParams:
|
||||||
self.options = options
|
self.options = options
|
||||||
|
|
||||||
def __str__(self) -> str:
|
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:
|
def torch_device(self) -> str:
|
||||||
if self.device.startswith('cuda'):
|
if self.device.startswith('cuda'):
|
||||||
|
|
Loading…
Reference in New Issue