[cm] Changing prepare for inference

Neutone · Feb 1, 2024 · b9dc93c · b9dc93c
1 parent 8c1b11a
commit b9dc93c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/neutone_sdk/conv.py b/neutone_sdk/conv.py
@@ -59,6 +59,7 @@ def reset(self, batch_size: Optional[int] = None) -> None:
     def prepare_for_inference(self) -> None:
         self.debug_mode = False
         self.reset()
+        self.eval()
 
     def forward(self, x: Tensor) -> Tensor:
         if self.debug_mode: