p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

mode_export_example

This commit is contained in:
bai-yangfan 2020-11-09 18:40:25 +08:00
parent 5887107264
commit a0a4f77524
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/lenet_quant/export.py
View File

@ -56,4 +56,4 @@ if __name__ == "__main__":
# export network
inputs = Tensor(np.ones([1, 1, cfg.image_height, cfg.image_width]), mindspore.float32)
export(network, inputs, file_name="lenet_quant", file_format='MINDIR', quant_mode='AUTO')
export(network, inputs, file_name="lenet_quant.mindir", file_format='MINDIR', quant_mode='AUTO')

2
model_zoo/official/cv/mobilenetv2_quant/export.py
View File

@ -53,5 +53,5 @@ if __name__ == '__main__':
# export network
print("============== Starting export ==============")
inputs = Tensor(np.ones([1, 3, cfg.image_height, cfg.image_width]), mindspore.float32)
export(network, inputs, file_name="mobilenet_quant", file_format='MINDIR', quant_mode='AUTO')
export(network, inputs, file_name="mobilenet_quant.mindir", file_format='MINDIR', quant_mode='AUTO')
print("============== End export ==============")

2
tests/st/quantization/lenet_quant/test_lenet_quant.py
View File

@ -147,7 +147,7 @@ def export_lenet():
# export network
inputs = Tensor(np.ones([1, 1, cfg.image_height, cfg.image_width]), mstype.float32)
export(network, inputs, file_name="lenet_quant", file_format='MINDIR', quant_mode='AUTO')
export(network, inputs, file_name="lenet_quant.mindir", file_format='MINDIR', quant_mode='AUTO')
@pytest.mark.level0