Skip to content

Commit 8c1b11a

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

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

neutone_sdk/conv.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,8 +268,8 @@ def prepare_for_inference(self) -> None:
268268
log.info(f"Converting Conv1dGeneral to cached in prepare_for_inference()")
269269
self.set_cached(True)
270270
self.debug_mode = False
271-
self.conv1d.eval()
272271
self.padding_cached.prepare_for_inference()
272+
self.eval()
273273

274274
def forward(self, x: Tensor) -> Tensor:
275275
"""

0 commit comments

Comments
 (0)