Update burn to 0.11.1 (#14)

This involves renaming the backend to the new structure
tracel-ai · Jan 30, 2024 · acaed68 · acaed68
1 parent 19b08b9
commit acaed68
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/squeezenet-burn/Cargo.toml b/squeezenet-burn/Cargo.toml
@@ -21,7 +21,7 @@ weights_file = ["burn/default"]
 [dependencies]
 
 # Note: default-features = false is needed to disable std
-burn = { version = "0.10.0", default-features = false }
+burn = { version = "0.11.1", default-features = false }
 
 # Used to load weights from a file
 serde = { version = "1.0.183", default-features = false, features = [
@@ -31,9 +31,9 @@ serde = { version = "1.0.183", default-features = false, features = [
 
 [dev-dependencies]
 # Used by the classify example
-burn-ndarray = { version = "0.10.0", package = "burn-ndarray" }
+burn = { version = "0.11.1", features = ["ndarray"]  }
 image = { version = "0.24.7", features = ["png", "jpeg"] }
 
 [build-dependencies]
 # Used to generate code from ONNX model
-burn-import = { version = "0.10.0", package = "burn-import" }
+burn-import = { version = "0.11.1", package = "burn-import" }
diff --git a/squeezenet-burn/examples/classify.rs b/squeezenet-burn/examples/classify.rs
@@ -1,7 +1,7 @@
 use squeezenet_burn::model::{label::LABELS, normalizer::Normalizer, squeezenet1::Model};
 
+use burn::backend::NdArray;
 use burn::tensor::Tensor;
-use burn_ndarray::NdArrayBackend;
 
 use image::{self, GenericImageView, Pixel};
 
@@ -11,7 +11,7 @@ const WIDTH: usize = 224;
 #[cfg(feature = "weights_file")]
 const RECORD_FILE: &str = "squeezenet1";
 
-type Backend = NdArrayBackend<f32>;
+type Backend = NdArray<f32>;
 
 fn main() {
     // Path to the image from the main args