feat: use static capture for squeezenet1_1 json

graph_net/samples/paddle_samples/vision/squeezenet1_1/graph_hash.txt

@@ -0,0 +1 @@
+6de9a1959b0f3ccd2b9e70f1b42a4f295af1bea89e4779c1e0fe5753bc609be7

graph_net/samples/paddle_samples/vision/squeezenet1_1/input_meta.py

@@ -0,0 +1,3 @@
+import paddle
+
+input_meta = {"inputs": {"shape": [1, 3, 224, 224], "dtype": "float32"}}

graph_net/samples/paddle_samples/vision/squeezenet1_1/model.py

@@ -0,0 +1,11 @@
+import paddle
+from paddle.vision.models import squeezenet1_1
+
+
+class GraphModule(paddle.nn.Layer):
+    def __init__(self):
+        super(GraphModule, self).__init__()
+        self.model = squeezenet1_1(pretrained=False)
+
+    def forward(self, inputs):
+        return self.model(inputs)

graph_net/samples/paddle_samples/vision/squeezenet1_1/weight_meta.py

@@ -0,0 +1,3 @@
+import paddle
+
+weight_meta = {}

graph_net/tests/test_squeezenet1_1_extract.py

@@ -0,0 +1,28 @@
+import paddle
+import os
+from paddle.vision.models import squeezenet1_1
+
+
+def extract_squeezenet():
+    # 环境准备
+    os.environ["FLAGS_enable_pir_api"] = "1"
+    paddle.enable_static()
+
+    # 捕获计算图
+    main_program = paddle.static.Program()
+    with paddle.static.program_guard(main_program):
+        inputs = paddle.static.data(
+            name="inputs", shape=[1, 3, 224, 224], dtype="float32"
+        )
+        model = squeezenet1_1(pretrained=False)
+        out = model(inputs)
+
+    # 保存结果
+    save_path = "squeezenet1_1.json"
+    with open(save_path, "w") as f:
+        f.write(str(main_program))
+    print(f"提取完成，文件已生成至: {save_path}")
+
+
+if __name__ == "__main__":
+    extract_squeezenet()