Set task for Triton inferenceTriton (#18064)
This commit is contained in:
parent
a672bf79dd
commit
f6005a9166
1 changed files with 1 additions and 0 deletions
|
|
@ -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…
Add table
Add a link
Reference in a new issue