Luigi commited on
Commit
01042ef
·
1 Parent(s): ae1686f

Remove fix_batch_dimension_all.sh

Browse files
Files changed (1) hide show
  1. fix_batch_dimension_all.sh +0 -20
fix_batch_dimension_all.sh DELETED
@@ -1,20 +0,0 @@
1
- #!/bin/bash
2
-
3
- # Loop through all .onnx files in the current directory
4
- for model in ./*.onnx; do
5
- # Exclude processing any files that already have a batch size in their name
6
- if [[ ! $model =~ _batch[0-9]+\.onnx$ ]]; then
7
- # Process for batch size 1
8
- python3 -m onnxruntime.tools.make_dynamic_shape_fixed --dim_param batch --dim_value 1 "$model" "${model%.onnx}_batch1.onnx"
9
- echo "Generated ${model%.onnx}_batch1.onnx"
10
-
11
- # Process for batch size 2
12
- python3 -m onnxruntime.tools.make_dynamic_shape_fixed --dim_param batch --dim_value 2 "$model" "${model%.onnx}_batch2.onnx"
13
- echo "Generated ${model%.onnx}_batch2.onnx"
14
-
15
- # Process for batch size 4
16
- python3 -m onnxruntime.tools.make_dynamic_shape_fixed --dim_param batch --dim_value 4 "$model" "${model%.onnx}_batch4.onnx"
17
- echo "Generated ${model%.onnx}_batch4.onnx"
18
- fi
19
- done
20
-