def default_data_to_device(
    input, target=None, device: str = "cuda", non_blocking: bool = True
):
    """Sends data output from a PyTorch Dataloader to the device."""

    input = input.to(device=device, non_blocking=non_blocking)

    if target is not None:
        target = target.to(device=device, non_blocking=non_blocking)

    return input, target