mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 10:38:54 +00:00
Training:
- Removed a lot of surface (SerializedFileReader ownership is really painful). - Moved example + vision to hf.co version. - Removed feature gate.
This commit is contained in:
@ -143,7 +143,7 @@ struct Args {
|
||||
pub fn main() -> anyhow::Result<()> {
|
||||
let args = Args::parse();
|
||||
// Load the dataset
|
||||
let m = candle_datasets::vision::mnist::load_dir("data")?;
|
||||
let m = candle_datasets::vision::mnist::load()?;
|
||||
println!("train-images: {:?}", m.train_images.shape());
|
||||
println!("train-labels: {:?}", m.train_labels.shape());
|
||||
println!("test-images: {:?}", m.test_images.shape());
|
||||
|
Reference in New Issue
Block a user