Skip to content

Commit b9dc93c

Browse files
committed
[cm] Changing prepare for inference
1 parent 8c1b11a commit b9dc93c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

neutone_sdk/conv.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ def reset(self, batch_size: Optional[int] = None) -> None:
5959
def prepare_for_inference(self) -> None:
6060
self.debug_mode = False
6161
self.reset()
62+
self.eval()
6263

6364
def forward(self, x: Tensor) -> Tensor:
6465
if self.debug_mode:

0 commit comments

Comments
 (0)