mirror of
https://github.com/huggingface/candle.git
synced 2025-06-21 12:20:46 +00:00
Add the candle-datasets crate (#322)
* Move the vision datasets to a separate crate. * Move the batcher bits. * Update the readme. * Move the tiny-stories bits. --------- Co-authored-by: Jane Doe <jane.doe@example.org>
This commit is contained in:
12
candle-datasets/src/vision/mod.rs
Normal file
12
candle-datasets/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