Skip to content
Snippets Groups Projects
Commit e34b58c0 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Shape

parent a832e239
No related branches found
No related tags found
No related merge requests found
...@@ -220,7 +220,7 @@ class SequenceNetwork(six.with_metaclass(abc.ABCMeta, object)): ...@@ -220,7 +220,7 @@ class SequenceNetwork(six.with_metaclass(abc.ABCMeta, object)):
hdf5.set('input_divide', self.input_divide) hdf5.set('input_divide', self.input_divide)
hdf5.set('input_subtract', self.input_subtract) hdf5.set('input_subtract', self.input_subtract)
def load(self, hdf5, shape, session=None): def load(self, hdf5, shape=None, session=None):
""" """
Load the network Load the network
...@@ -244,6 +244,9 @@ class SequenceNetwork(six.with_metaclass(abc.ABCMeta, object)): ...@@ -244,6 +244,9 @@ class SequenceNetwork(six.with_metaclass(abc.ABCMeta, object)):
self.sequence_net = pickle.loads(hdf5.read('architecture')) self.sequence_net = pickle.loads(hdf5.read('architecture'))
self.deployment_shape = hdf5.read('deployment_shape') self.deployment_shape = hdf5.read('deployment_shape')
if shape is None:
shape = self.deployment_shape
# Loading variables # Loading variables
place_holder = tf.placeholder(tf.float32, shape=shape, name="load") place_holder = tf.placeholder(tf.float32, shape=shape, name="load")
self.compute_graph(place_holder) self.compute_graph(place_holder)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment