Merge pull request #5 from antimora/normalizer-fix

Fix model normalizer
tracel-ai · Oct 9, 2023 · 99be61c · 99be61c
2 parents 5323c22 + 8ef8bfa
commit 99be61c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/squeezenet-burn/src/model/normalizer.rs b/squeezenet-burn/src/model/normalizer.rs
@@ -26,8 +26,8 @@ impl<B: Backend> Normalizer<B> {
     ///
     /// The normalization is done according to the following formula:
     /// `input = (input - mean) / std`
-    pub fn normalize(self, input: Tensor<B, 4>) -> Tensor<B, 4> {
-        input - self.mean / self.std
+    pub fn normalize(&self, input: Tensor<B, 4>) -> Tensor<B, 4> {
+        (input - self.mean.clone()) / self.std.clone()
     }
 }