diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 6d403a2afb..1c049d02de 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -167,7 +167,7 @@ class Exporter: self.callbacks = _callbacks or callbacks.get_default_callbacks() callbacks.add_integration_callbacks(self) - @smart_inference_mode() + # @smart_inference_mode() def __call__(self, model=None) -> str: """Returns list of exported files/dirs after running callbacks.""" self.run_callbacks("on_export_start") diff --git a/ultralytics/nn/autobackend.py b/ultralytics/nn/autobackend.py index cc812b8332..0dcbb12f96 100644 --- a/ultralytics/nn/autobackend.py +++ b/ultralytics/nn/autobackend.py @@ -79,6 +79,7 @@ class AutoBackend(nn.Module): models across various platforms. """ + @torch.no_grad() def __init__( self, weights="yolo11n.pt",