class_docstr: Adds PyTorch specific functions.
class_docstr_tok: ["Adds","PyTorch","specific","functions","."]
docstr: Move model to CUDA.
docstr_tok: ["Move","model","to","CUDA","."]
code: def __move_model__(self):
      """Move model to CUDA."""
      # Setup cuda - do this before setting up optimizers!
      if not self.base.cuda:
          self.cuda = False
      elif self.base.cuda:
          if not torch.cuda.is_available():
              raise ValueError("Cuda is not available")
          self.cuda = True
          self.device = torch.device(self.base.cuda)
  
      if self.cuda:
          self.adversary_model = self.adversary_model.to(self.device)
          self.predictor_model = self.predictor_model.to(self.device)
  
file_path: fairlearn/fairlearn/adversarial/_pytorch_engine.py
imports: from ._backend_engine import BackendEngine
  from ._constants import _MODEL_UNRECOGNIZED_STR, _MODEL_UNRECOGNIZED_ITEM
label_desc: Use image node_id 2 for calling the PytorchEngine obj's underlying member method code with example usage: obj.__move_model__() without return types