diff --git a/model_zoo/official/cv/lenet_quant/export.py b/model_zoo/official/cv/lenet_quant/export.py index 4f5d9381d2d..f144ce59364 100644 --- 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') diff --git a/model_zoo/official/cv/mobilenetv2_quant/export.py b/model_zoo/official/cv/mobilenetv2_quant/export.py index 735673415fe..5dec710525b 100644 --- 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 ==============") diff --git a/tests/st/quantization/lenet_quant/test_lenet_quant.py b/tests/st/quantization/lenet_quant/test_lenet_quant.py index 6bde63a112d..728a58943c8 100644 --- 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