mode_export_example

4 years ago · a0a4f77524
parent 5887107264
commit a0a4f77524
3 changed files with 3 additions and 3 deletions
--- a/model_zoo/official/cv/lenet_quant/export.py
+++ b/model_zoo/official/cv/lenet_quant/export.py
@ -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')
--- a/model_zoo/official/cv/mobilenetv2_quant/export.py
+++ b/model_zoo/official/cv/mobilenetv2_quant/export.py
@ -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 ==============")
--- a/tests/st/quantization/lenet_quant/test_lenet_quant.py
+++ b/tests/st/quantization/lenet_quant/test_lenet_quant.py
@ -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