Skip to content

Commit c4aa969

Browse files
committed
update onnx model params
Signed-off-by: yuwenzho <[email protected]>
1 parent a384d85 commit c4aa969

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/.config/model_params_onnxrt.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -872,14 +872,14 @@
872872
"model_src_dir": "object_detection/table_transformer/quantization/ptq_static",
873873
"dataset_location": "/tf_dataset/dataset/PubTables-1M",
874874
"input_model": "/tf_dataset2/models/onnx/table-transformer/pubtables1m_structure_detr_r18.onnx",
875-
"main_script": "table-transformer/src/main.py",
875+
"main_script": "patch",
876876
"batch_size": 1
877877
},
878878
"table_transformer_detection": {
879879
"model_src_dir": "object_detection/table_transformer/quantization/ptq_static",
880880
"dataset_location": "/tf_dataset/dataset/PubTables-1M",
881881
"input_model": "/tf_dataset2/models/onnx/table-transformer/pubtables1m_detection_detr_r18.onnx",
882-
"main_script": "table-transformer/src/main.py",
882+
"main_script": "patch",
883883
"batch_size": 1
884884
},
885885
"hf_codebert": {

0 commit comments

Comments
 (0)