Skip to content

Commit 30cb6c5

Browse files
Comment out the torch.compile line in ModelnnUNetWrapper to prevent potential issues during execution
Signed-off-by: simben <[email protected]>
1 parent 1032c9b commit 30cb6c5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

monai/apps/nnunet/nnunet_bundle.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -252,7 +252,7 @@ def __init__(self, predictor, model_folder, model_name="model.pt"):
252252
# and not isinstance(predictor.network, OptimizedModule)
253253
):
254254
print("Using torch.compile")
255-
predictor.network = torch.compile(self.network)
255+
# predictor.network = torch.compile(self.network)
256256
# End Block
257257
self.network_weights = self.predictor.network
258258

0 commit comments

Comments
 (0)