diff options
Diffstat (limited to 'candle-examples/examples/simple-training/main.rs')
| -rw-r--r-- | candle-examples/examples/simple-training/main.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/simple-training/main.rs b/candle-examples/examples/simple-training/main.rs index 2cfe4923..60f2281b 100644 --- a/candle-examples/examples/simple-training/main.rs +++ b/candle-examples/examples/simple-training/main.rs @@ -52,7 +52,7 @@ pub fn main() -> Result<()> { .to_dtype(DType::F32)? .sum_all()? .to_scalar::<f32>()?; - let test_accuracy = sum_ok / test_labels.shape().r1()? as f32; + let test_accuracy = sum_ok / test_labels.dims1()? as f32; println!( "{epoch:4} train loss: {:8.5} test acc: {:5.2}%", loss.to_scalar::<f32>()?, |