administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-25 16:17:41 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
Nicolas Mowen 2025-09-17 17:35:32 -06:00
parent 33042592c7
commit 78ac699610
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/detectors/plugins/onnx.py
View File

@ -56,7 +56,9 @@ class ONNXDetector(DetectionApi):
if providers[0] == "ROCMExecutionProvider": if providers[0] == "ROCMExecutionProvider":
# avoid AMD GPU kernel crashes # avoid AMD GPU kernel crashes
sess_options = ort.SessionOptions() sess_options = ort.SessionOptions()
sess_options.graph_optimization_level = ort.GraphOptimizationLevel.ORT_DISABLE_ALL sess_options.graph_optimization_level = (
ort.GraphOptimizationLevel.ORT_DISABLE_ALL
)
self.model = ort.InferenceSession( self.model = ort.InferenceSession(
path, providers=providers, provider_options=options path, providers=providers, provider_options=options