code: def device(
      self, __device: _typing.Union[torch.device, str, int, None]
  ):
      if (
          type(__device) == torch.device
          or (type(__device) == str and __device.lower() != "auto")
          or type(__device) == int
      ):
          self._device: torch.device = torch.device(__device)
      else:
          self._device: torch.device = torch.device(
              "cuda"
              if torch.cuda.is_available()
              and torch.cuda.device_count() > 0
              else "cpu"
          )
  
file_path: AutoGL/autogl/module/hpo/auto_module.py
imports: import torch
  import typing
label_desc: Use image node_id 5 for calling the AutoModule obj's underlying member method code with example usage: obj.device(__device) without return types