We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a9c3f9a commit fbabcadCopy full SHA for fbabcad
keras2ncnn/graph_optimizer.py
@@ -7,7 +7,10 @@ def removing_unused_nodes(graph):
7
8
for target_node_name in graph.get_graph().keys():
9
if graph.get_node_attr(target_node_name)[
10
- 'layer']['class_name'] in UNUSED_NODES:
+ 'layer']['class_name'] in UNUSED_NODES or \
11
+ ( graph.get_node_attr(target_node_name)[
12
+ 'layer']['class_name'] == 'InputLayer' and \
13
+ len(graph.get_node_inbounds(target_node_name)) != 0 ):
14
for layer_name in graph.get_graph().keys():
15
if target_node_name in graph.get_graph()[
16
layer_name]['inbounds']:
0 commit comments