mirror of
https://github.com/azaion/detections.git
synced 2026-06-21 08:41:07 +00:00
Fixed pipeline
This commit is contained in:
@@ -65,7 +65,7 @@ services:
|
||||
- jetson
|
||||
build:
|
||||
context: ..
|
||||
dockerfile: Dockerfile.jetson
|
||||
dockerfile: ${JETSON_DOCKERFILE:-Dockerfile.jetson}
|
||||
runtime: nvidia
|
||||
depends_on:
|
||||
- mock-loader
|
||||
|
||||
@@ -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