mirror of
https://github.com/azaion/detections.git
synced 2026-06-21 08:11:07 +00:00
Fixed pipeline
This commit is contained in:
@@ -28,6 +28,15 @@ if [[ "$engine_count" == "0" ]]; then
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [[ "${REQUIRE_INT8_ENGINE:-0}" == "1" ]]; then
|
||||
int8_engine_count="$(find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.int8.engine' | wc -l | tr -d ' ')"
|
||||
if [[ "$int8_engine_count" == "0" ]]; then
|
||||
echo "ERROR: INT8 engine is required, but no azaion*.int8.engine file was produced"
|
||||
find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.engine' -print -exec ls -lh {} \;
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
|
||||
echo "--- Converted TensorRT engine files:"
|
||||
find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.engine' -print -exec ls -lh {} \;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user