Skip to content

Commit 2092f79

Browse files
committed
[Feature Enhancement] Add 'dynamic' metadata to graph_net.json
1 parent a342d13 commit 2092f79

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

graph_net/torch/extractor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ def try_rename_placeholder(node):
6666
"framework": "torch",
6767
"num_devices_required": 1,
6868
"num_nodes_required": 1,
69+
"dynamic": bool(dynamic),
6970
}
7071
with open(os.path.join(model_path, "graph_net.json"), "w") as f:
7172
json.dump(metadata, f, indent=4)

0 commit comments

Comments
 (0)