Skip to content

Commit 9f34d5d

Browse files
committed
[cm] Changing set_cached
1 parent 306abb3 commit 9f34d5d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

neutone_sdk/tcn.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -341,6 +341,7 @@ def prepare_for_inference(self) -> None:
341341
self.debug_mode = False
342342
for block in self.blocks:
343343
block.prepare_for_inference()
344+
self.set_cached(True)
344345
self.eval()
345346

346347
def forward(self, x: Tensor, cond: Optional[Tensor] = None) -> Tensor:

0 commit comments

Comments
 (0)