mirror of
https://github.com/huggingface/candle.git
synced 2025-06-20 20:09:50 +00:00
620f83cf66
* 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>
13 lines
216 B
Rust
13 lines
216 B
Rust
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;
|