bug fix

tracel-ai · May 6, 2024 · 9af7481 · 9af7481
1 parent 4297044
commit 9af7481
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 221 deletions.
diff --git a/crates/burn-import/src/burn/node/max_pool1d.rs b/crates/burn-import/src/burn/node/max_pool1d.rs
@@ -133,7 +133,7 @@ mod tests {
                 #[allow(unused_variables)]
                 pub fn new(device: &B::Device) -> Self {
                     let max_pool1d = MaxPool1dConfig::new(3)
-                        .with_strides(1)
+                        .with_stride(1)
                         .with_padding(PaddingConfig1d::Valid)
                         .with_dilation(1)
                         .init();

diff --git a/flake.lock b/flake.lock
diff --git a/flake.nix b/flake.nix