mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-10-08 15:52:48 +03:00
Update ROCm to not hang when running on complex RNN models (#20118)
* Update ROCm to not hang when running on complex RNN models * Formatting
This commit is contained in:
@@ -51,6 +51,15 @@ class ONNXDetector(DetectionApi):
|
||||
"enable_cuda_graph": True,
|
||||
}
|
||||
|
||||
sess_options = None
|
||||
|
||||
if providers[0] == "ROCMExecutionProvider":
|
||||
# avoid AMD GPU kernel crashes
|
||||
sess_options = ort.SessionOptions()
|
||||
sess_options.graph_optimization_level = (
|
||||
ort.GraphOptimizationLevel.ORT_DISABLE_ALL
|
||||
)
|
||||
|
||||
self.model = ort.InferenceSession(
|
||||
path, providers=providers, provider_options=options
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user