Skip to content

Commit 17fe33e

Browse files
authored
Merge pull request #403 from bioimage-io/torchscript_eval
Make sure torchscript models run in eval mode
2 parents ce97def + 0f9d7a0 commit 17fe33e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

bioimageio/core/model_adapters/_torchscript_model_adapter.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ def __init__(
5050

5151
self._model = torch.jit.load(weight_path)
5252
self._model.to(self.devices[0])
53+
self._model = self._model.eval()
5354
self._internal_output_axes = [
5455
tuple(a.id for a in get_axes_infos(out))
5556
for out in model_description.outputs

0 commit comments

Comments
 (0)