(self, device)
Set the `device` to be used ('GPU' or 'CPU).
Set the `device` to be used ('GPU' or 'CPU).
Set the device to be used ( GPU or CPU ) .
def set_device(self, device):
        """Set the `device` to be used ('GPU' or 'CPU)."""
        self._device = device
        if self._generator is not None:
            self._generator.to(self._device)
def set_device ( self , device ) : self . _device = device if self . _generator is not None : self . _generator . to ( self . _device )