You can try exporting to TensorRT
with torch.no_grad() shouldn’t be used with Ultralytics. And model.to() should also be removed because Ultralytics handles that.
Is your device a desktop or a laptop?
You can try exporting to TensorRT
with torch.no_grad() shouldn’t be used with Ultralytics. And model.to() should also be removed because Ultralytics handles that.
Is your device a desktop or a laptop?