test remove smart inference

no-grad-test
Francesco Mattioli 3 months ago
parent df70f72921
commit 5797620ad5
  1. 2
      ultralytics/engine/exporter.py
  2. 1
      ultralytics/nn/autobackend.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")

@ -79,6 +79,7 @@ class AutoBackend(nn.Module):
models across various platforms.
"""
@torch.no_grad()
def __init__(
self,
weights="yolo11n.pt",

Loading…
Cancel
Save