Skip to content

misc models

Guenther Schmuelling edited this page Oct 15, 2020 · 7 revisions

albert, albert-lite

Url

https://tfhub.dev/tensorflow/albert_lite_base/1
https://tfhub.dev/google/albert_base/3

Command

python -m tf2onnx.convert  --saved-model albert --output albert/albert_lite_base_squadv1_1.onnx --opset 11  --signature_def tokens --tag ''

mobilebert

Url

https://github.com/google-research/google-research/tree/master/mobilebert

Command

python -m tf2onnx.convert --saved-model float --output mobilebert.float.onnx --tag serve --opset 11

ssd-mobilenet-v2

Url

http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v2_coco_2018_03_29.tar.gz

Command

python -m tf2onnx.convert --saved-model D:\MLPerf\models\ssd_mobilenet_v2\tf\saved_model --opset 11 --output ssd_mobilenet_v2.onnx

MobileNetEdgeTPU

Url

https://zenodo.org/record/3906875#.XvUgoi05TUJ

Command

python -m tf2onnx.convert --opset 11 --input mobilenetedgetpu_fp32.pb  --output mobilenet_edgetpu_224_1.0.onnx --inputs input:0 --outputs MobilenetEdgeTPU/Predictions/Reshape_1:0