fix stupid

2026-03-26 10:59:02 -07:00
parent 4d3811a590
commit 02cfd6c052
1 changed files with 2 additions and 1 deletions
--- a/src/main.rs
+++ b/src/main.rs
@@ -48,8 +48,9 @@ impl<B: Backend> Net<B> {
 		let input = LinearConfig::new(INPUT, last_size).init(device);
 		let hidden = core::array::from_fn(|_| {
 			let size = it.next().unwrap();
+			let layer = LinearConfig::new(last_size, size).init(device);
 			last_size = size;
-			LinearConfig::new(last_size, size).init(device)
+			layer
 		});
 		let output = LinearConfig::new(last_size, OUTPUT).init(device);
 		Self {