Vision dataset (#179)

* Add some readers for the mnist dataset.

* Import the cifar and mnist dataset.
This commit is contained in:
Laurent Mazare
2023-07-16 23:43:55 +01:00
committed by GitHub
parent 6de7345e39
commit 2a74019ec6
4 changed files with 140 additions and 0 deletions

View File

@ -0,0 +1,12 @@
use candle::Tensor;
pub struct Dataset {
pub train_images: Tensor,
pub train_labels: Tensor,
pub test_images: Tensor,
pub test_labels: Tensor,
pub labels: usize,
}
pub mod cifar;
pub mod mnist;