mirror of
https://github.com/huggingface/candle.git
synced 2025-06-20 12:06:35 +00:00
Vision dataset (#179)
* Add some readers for the mnist dataset. * Import the cifar and mnist dataset.
This commit is contained in:
12
candle-nn/src/vision/mod.rs
Normal file
12
candle-nn/src/vision/mod.rs
Normal 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;
|
Reference in New Issue
Block a user