Set `task` for Triton inferenceTriton (#18064)

pull/18059/head
Laughing 7 days ago committed by GitHub
parent a672bf79dd
commit f6005a9166
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 1
      ultralytics/engine/model.py

@ -136,6 +136,7 @@ class Model(nn.Module):
# Check if Triton Server model
elif self.is_triton_model(model):
self.model_name = self.model = model
self.overrides["task"] = task or "detect" # set `task=detect` if not explicitly set
return
# Load or create new YOLO model

Loading…
Cancel
Save