refactor: update Cargo.toml dependencies and add recursion limit in main.rs

SammyOina · SammyOina · commit 334704f3ced6 · 2025-10-21T20:33:42.000+03:00
Signed-off-by: Sammy Oina &lt;sammyoina@gmail.com&gt;
diff --git a/burn-algorithms/imdb/Cargo.toml b/burn-algorithms/imdb/Cargo.toml
@@ -28,7 +28,9 @@ required-features = ["cocos"]
 [dependencies]
 burn = { version = "0.18.0", features = [
     "dataset",
+    "ndarray",
     "train",
+    "wgpu",
 ] }
 csv = "1.3.0"
 derive-new = "0.7.0"
diff --git a/burn-algorithms/imdb/src/main.rs b/burn-algorithms/imdb/src/main.rs
@@ -1,3 +1,5 @@
+#![recursion_limit = "256"]
+
 use burn::{
     nn::transformer::TransformerEncoderConfig,
     optim::{decay::WeightDecayConfig, AdamConfig},

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,5 @@`
	`1`	`+#![recursion_limit = "256"]`
	`2`	`+`
`1`	`3`	`use burn::{`
`2`	`4`	`nn::transformer::TransformerEncoderConfig,`
`3`	`5`	`optim::{decay::WeightDecayConfig, AdamConfig},`