🐛 Fix wrong intput to compile_model.

TissueImageAnalytics · Nov 17, 2024 · e88c125 · e88c125
1 parent 035ab74
commit e88c125
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tiatoolbox/models/engine/engine_abc.py b/tiatoolbox/models/engine/engine_abc.py
@@ -358,7 +358,7 @@ def __init__(
         self.model.to(device=self.device)
         self.model = (
             compile_model(  # for runtime, such as after wrapping with nn.DataParallel
-                model,
+                self.model,
                 mode=rcParam["torch_compile_mode"],
             )
         )