HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Set task for Triton inferenceTriton (#18064)

Browse source
This commit is contained in:
Laughing 2024-12-06 18:46:48 +08:00 • committed by GitHub
parent a672bf79dd
commit f6005a9166
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
ultralytics/engine/model.py
View file

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