Skip to content

Commit 081f898

Browse files
author
Alex
committed
Merge branch 'release/1.9.3'
2 parents b62fe07 + 47523d7 commit 081f898

File tree

7 files changed

+276
-233
lines changed

7 files changed

+276
-233
lines changed

backend/translate/graph.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,6 @@ class Graph:
1111
"""Representing a Network Graph."""
1212
def __init__(self):
1313
self.layers = []
14-
self.root = Node()
15-
self.input_shape = []
1614

1715

1816
def add_layer(self, layer):
@@ -23,13 +21,6 @@ def add_layer(self, layer):
2321
"""
2422
self.layers.append(layer)
2523

26-
def set_input_shape(self, input_shape):
27-
"""Set the input shape of the model.
28-
29-
Arguments:
30-
input_shape {list} -- the input shape of the model
31-
"""
32-
self.input_shape = input_shape
3324

3425
# Return the Graph representation.
3526
def __repr__(self):

backend/translate/layer.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,9 @@ def add_input_names(self, nodes):
4343
Arguments:
4444
nodes {list} -- all nodes that are inputs to the layer
4545
"""
46-
for node in nodes[0]:
47-
self.input_names.append(node[0])
46+
if (len(nodes) > 0):
47+
for node in nodes[0]:
48+
self.input_names.append(node[0])
4849

4950
# String Representation of the Layer.
5051
def __repr__(self):

backend/translate/translate_keras.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,6 @@ def graph_from_external_file(keras_code, general_code):
5151
model_keras = keras.models.model_from_json(result["stdout"])
5252
layers_extracted = model_json['config']['layers']
5353
graph = Graph()
54-
graph.set_input_shape(
55-
layers_extracted[0]['config']['batch_input_shape'][1:])
5654
previous_node = ''
5755
for index, json_layer in enumerate(layers_extracted):
5856
if len(layers_extracted) > len(model_keras.layers):

0 commit comments

Comments
 (0)