diff --git a/neutone_sdk/conv.py b/neutone_sdk/conv.py
index 340c535..62ff0fb 100644
--- 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: