def METHOD_NAME(self, device, available_devices=1):
    """
    Changes device of model to support CPU/GPU
    :param device: will move the model to this device.
    :param available_devices: all available devices as reported by lightwood.
    :return: same object but moved to the target device.
    """
    self.model.METHOD_NAME(device, available_devices=1)
    return self