candle icon indicating copy to clipboard operation
candle copied to clipboard

Cannot download CIFAR-10

Open VillSnow opened this issue 1 year ago • 5 comments

fn main() -> Result<(), Box<dyn Error>> {
    let cifar = candle_datasets::vision::cifar::load()?;

    dbg!(&cifar.train_images); // Tensor[dims 50000, 3, 32, 32; u8]
    dbg!(&cifar.train_labels); // Tensor[dims 50000; u8]

    dbg!(cifar
        .train_images
        .flatten_all()?
        .to_vec1::<u8>()?
        .into_iter()
        .any(|x| x != 0)); // false

    dbg!(cifar
        .train_labels
        .flatten_all()?
        .to_vec1::<u8>()?
        .into_iter()
        .any(|x| x != 0)); // true

    Ok(())
}

I was able to see download progress bars for plain_text/test/0000.parquet and lain_text/train/0000.parquet, but no for images.

VillSnow avatar Nov 19 '24 13:11 VillSnow