Add burn/ndarry to dev-dependencies

tracel-ai · Feb 14, 2024 · 20b1878 · 20b1878
1 parent e624b65
commit 20b1878
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/resnet-burn/Cargo.toml b/resnet-burn/Cargo.toml
@@ -21,4 +21,7 @@ serde = { version = "1.0.192", default-features = false, features = [
 ] } # alloc is for no_std, derive is needed
 
 [dev-dependencies]
+burn = { git = "https://github.com/tracel-ai/burn.git", rev = "9a2cbadd41161c8aac142bbcb9c2ceaf5ffd6edd", features = [
+    "ndarray",
+] }
 image = { version = "0.24.7", features = ["png", "jpeg"] }