def toDevice(datas, device):
    """
    Enable cuda.

    Args:
        datas: tensor
        device: cpu or cuda

    Returns:
        Transform `data` to `device`
    """
    imgs, angles = datas
    return imgs.float().to(device), angles.float().to(device)