ls_mlkit.util.base_config_class module

class ls_mlkit.util.base_config_class.BaseConfigClass(*args: list[Any], **kwargs: dict[Any, Any])[source]

Bases: object

to(device: device | str | Tensor, inplace: bool = True) BaseConfigClass[source]

Move the config to the given device

Parameters:
  • device (torch.device | str | Tensor) – the device to move the config to

  • inplace (bool, optional) – whether to move the config in place. Defaults to True.

Returns:

the config moved to the given device

Return type:

BaseConfigClass