Add AMD support via ort alternative to trt
This commit is contained in:
+4
-4
@@ -1,5 +1,5 @@
|
||||
#pragma once
|
||||
#include "ort_provider.hpp"
|
||||
#include "inference/backend_config.hpp"
|
||||
#include <string>
|
||||
|
||||
inline const std::string kDefaultDetectorModel = std::string(SAE_MODELS_DIR) + "/scrfd_500m_bnkps.onnx";
|
||||
@@ -58,10 +58,10 @@ struct Config {
|
||||
double extinction_sec{5.0}; // keep actor active this many seconds after last detection
|
||||
double anneal_sec{2.0}; // merge actor windows separated by less than this into one epoch
|
||||
|
||||
// ── TensorRT ──────────────────────────────────────────────────────────────
|
||||
// Only active when OrtProvider::TensorRT is detected.
|
||||
// ── Inference backend tuning ────────────────────────────────────────────
|
||||
// Consumed by the compiled-in inference backend (ORT or TRT).
|
||||
// INT8 is unsafe for ArcFace without a calibration table.
|
||||
TrtConfig trt{}; // fp16=true, int8=false, cache_dir="./trt_cache"
|
||||
BackendConfig trt{}; // fp16=true, int8=false, cache_dir="./trt_cache"
|
||||
|
||||
// ── Debug output (only used when SAE_DEBUG is defined) ───────────────────
|
||||
#ifdef SAE_DEBUG
|
||||
|
||||
Reference in New Issue
Block a user