curobo.types.base module

class curobo.types.base.TensorDeviceType(device: torch.device = device(type='cuda', index=0), dtype: torch.float32 = torch.float32)

Bases: object

Parameters:
  • device (device) –

  • dtype (torch.float32) –

device: device = device(type='cuda', index=0)
dtype: torch.float32 = torch.float32
static from_basic(device, dev_id)
Parameters:
  • device (str) –

  • dev_id (int) –

to_device(data_tensor)
to_int8_device(data_tensor)
cpu()