format devices in logs
This commit is contained in:
parent
a212d954eb
commit
2c9a10e2b7
|
@ -65,6 +65,9 @@ class DeviceParams:
|
||||||
self.provider = provider
|
self.provider = provider
|
||||||
self.options = options
|
self.options = options
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return '%s - %s' % (self.device, self.provider)
|
||||||
|
|
||||||
def torch_device(self) -> str:
|
def torch_device(self) -> str:
|
||||||
if self.device.startswith('cuda'):
|
if self.device.startswith('cuda'):
|
||||||
return self.device
|
return self.device
|
||||||
|
|
Loading…
Reference in New Issue