diff --git a/x2paddle/onnx_infer.py b/x2paddle/onnx_infer.py index b883a254bb46160c89d34d3c3604a8dd996a0514..a845f7052adc7b1a05bead5438aa238fd8aedf47 100644 --- a/x2paddle/onnx_infer.py +++ b/x2paddle/onnx_infer.py @@ -48,5 +48,6 @@ def main(): for idx, value_info in enumerate(model.graph.output): np.save(os.path.join(save_dir, value_info.name), res[idx]) + if __name__ == "__main__": main()