Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Empty file.
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
2f1fbe960d81387133cc93e530c7efa3defdb56c134a0253689cfa85f297331e
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
import paddle

input_meta = {"inputs": {"shape": [1, 3, 224, 224], "dtype": "float32"}}
11 changes: 11 additions & 0 deletions graph_net/samples/paddle_samples/vision/squeezenet1_1/model.py
Original file line number Diff line number Diff line change
@@ -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)

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
import paddle

weight_meta = {}
20 changes: 20 additions & 0 deletions graph_net/tests/test_squeezenet1_1_extract.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
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)
model(inputs)

save_path = "graph_net/samples/paddle_samples/vision/squeezenet1_1/squeezenet1_1.json"
with open(save_path, "w") as f:
f.write(str(main_program))
print("Extract success")

if __name__ == "__main__":
extract_squeezenet()
Loading