Skip to content

Commit 49dbb5d

Browse files
DCO Remediation Commit for simben <[email protected]>
I, simben <[email protected]>, hereby add my Signed-off-by to this commit: 13bdec5 I, simben <[email protected]>, hereby add my Signed-off-by to this commit: 74aaf73 I, simben <[email protected]>, hereby add my Signed-off-by to this commit: 8e4a66c Signed-off-by: simben <[email protected]>
1 parent 8e4a66c commit 49dbb5d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

monai/bundle/nnunet.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616

1717
import numpy as np
1818
import torch
19-
from torch._dynamo import OptimizedModule
2019
from torch.backends import cudnn
2120

2221
from monai.data.meta_tensor import MetaTensor
@@ -249,7 +248,7 @@ def __init__(self, predictor, model_folder, model_name="model.pt"):
249248
if (
250249
("nnUNet_compile" in os.environ.keys())
251250
and (os.environ["nnUNet_compile"].lower() in ("true", "1", "t"))
252-
and not isinstance(predictor.network, OptimizedModule)
251+
#and not isinstance(predictor.network, OptimizedModule)
253252
):
254253
print("Using torch.compile")
255254
predictor.network = torch.compile(self.network)

0 commit comments

Comments
 (0)