From 0d4097031cb741e982524b7adccb8811287b1c29 Mon Sep 17 00:00:00 2001 From: Kyle Birnbaum Date: Wed, 26 Mar 2025 00:10:03 -0700 Subject: [PATCH] fixed rand import for mnist-training (#2833) --- candle-examples/examples/mnist-training/main.rs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/candle-examples/examples/mnist-training/main.rs b/candle-examples/examples/mnist-training/main.rs index a41a6496..097e13ee 100644 --- a/candle-examples/examples/mnist-training/main.rs +++ b/candle-examples/examples/mnist-training/main.rs @@ -7,6 +7,7 @@ extern crate accelerate_src; use clap::{Parser, ValueEnum}; use rand::prelude::*; +use rand::rng; use candle::{DType, Result, Tensor, D}; use candle_nn::{loss, ops, Conv2d, Linear, Module, ModuleT, Optimizer, VarBuilder, VarMap}; @@ -138,7 +139,7 @@ fn training_loop_cnn( let mut batch_idxs = (0..n_batches).collect::>(); for epoch in 1..args.epochs { let mut sum_loss = 0f32; - batch_idxs.shuffle(&mut thread_rng()); + batch_idxs.shuffle(&mut rng()); for batch_idx in batch_idxs.iter() { let train_images = train_images.narrow(0, batch_idx * BSIZE, BSIZE)?; let train_labels = train_labels.narrow(0, batch_idx * BSIZE, BSIZE)?;