Skip to content

Commit 2184735

Browse files
authored
remove --workspace option (for supporting latest trtexec)
1 parent 7df9eae commit 2184735

File tree

1 file changed

+2
-9
lines changed

1 file changed

+2
-9
lines changed

weights/tensorrt/convert.bash

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -2,22 +2,15 @@
22

33
# if $1 is empty
44
if [ -z "$1" ]; then
5-
echo "Usage: $0 <target-model> <workspace>"
5+
echo "Usage: $0 <target-model>"
66
echo "Target-Models : yolox_tiny, yolox_nano, yolox_s, yolox_m, yolox_l"
7-
echo "WORKSPACE : GPU memory workspace. Default 16."
87
exit 1
98
fi
109

1110
MODEL=$1
12-
TRT_WORKSPACE=$2
13-
if [ -z "$2" ]; then
14-
TRT_WORKSPACE=16
15-
fi
16-
1711
SCRIPT_DIR=$(cd $(dirname $0); pwd)
1812

1913
echo "Model Name: ${MODEL}"
20-
echo "Workspace size: ${TRT_WORKSPACE}"
2114
echo ""
2215

2316
ONNX_MODEL_PATH=$SCRIPT_DIR/../onnx/$MODEL.onnx
@@ -34,4 +27,4 @@ fi
3427
/usr/src/tensorrt/bin/trtexec \
3528
--onnx=$SCRIPT_DIR/../onnx/$MODEL.onnx \
3629
--saveEngine=$SCRIPT_DIR/$MODEL.trt \
37-
--fp16 --verbose --workspace=$((1<<$TRT_WORKSPACE))
30+
--fp16 --verbose

0 commit comments

Comments
 (0)