From 6fdc30c5cb850d68dfc46c7338060688a2627399 Mon Sep 17 00:00:00 2001 From: hebangwen <32662175+hebangwen@users.noreply.github.com> Date: Wed, 29 Jul 2026 15:09:25 +0800 Subject: [PATCH] bugfix: set correct workspace size parameter in build_engines.sh The `memPoolSize` parameter in trtexec uses `M` instead of `MiB` --- MiniCPM-RobotTrack/scripts/build_engines.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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"