diff --git a/MiniCPM-RobotTrack/scripts/build_engines.sh b/MiniCPM-RobotTrack/scripts/build_engines.sh index eb99bb7..a04376e 100644 --- a/MiniCPM-RobotTrack/scripts/build_engines.sh +++ b/MiniCPM-RobotTrack/scripts/build_engines.sh @@ -19,14 +19,14 @@ echo "Build engines only after MAXN mode 0 and jetson_clocks are active." "$TRTEXEC" \ --onnx="$ARTIFACTS/dino_patch_jp6_op17.onnx" \ --saveEngine="$ARTIFACTS/dino_patch_target_fp16.engine" \ - --fp16 --memPoolSize=workspace:4096MiB \ + --fp16 --memPoolSize=workspace:4096M \ --builderOptimizationLevel=5 --skipInference \ 2>&1 | tee "$LOGS/build-dino.log" "$TRTEXEC" \ --onnx="$ARTIFACTS/siglip_pooled_jp6.onnx" \ --saveEngine="$ARTIFACTS/siglip_pooled_target_maxn_fp16.engine" \ - --fp16 --memPoolSize=workspace:4096MiB \ + --fp16 --memPoolSize=workspace:4096M \ --builderOptimizationLevel=5 --skipInference \ 2>&1 | tee "$LOGS/build-siglip.log"