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

Added test case LightCNN9

parent 5f92d884
No related branches found
No related tags found
No related merge requests found
Pipeline #
......@@ -94,6 +94,11 @@ class MaxOut(base.Layer):
for i in range(len(shape)):
if shape[i] is None:
shape[i] = gen_array_ops.shape(inputs)[i]
outputs = math_ops.reduce_max(gen_array_ops.reshape(inputs, shape), -1, keep_dims=False)
shape = outputs.get_shape().as_list()
shape[self.axis] = self.num_units
outputs.set_shape(shape)
return outputs
......@@ -144,12 +144,12 @@ class LightCNN9(object):
reuse=reuse)
graph = maxout(graph,
num_units=256,
scope='Maxoutfc1')
name='Maxoutfc1')
graph = slim.fully_connected(graph, self.n_classes,
weights_initializer=initializer,
activation_fn=None,
scope='fc1',
scope='fc2',
reuse=reuse)
return graph
......@@ -123,7 +123,7 @@ def test_cnn_trainer():
del trainer
del graph
"""
def test_lightcnn_trainer():
# generating fake data
......@@ -158,7 +158,7 @@ def test_lightcnn_trainer():
# One graph trainer
trainer = Trainer(train_data_shuffler,
iterations=50,
iterations=5,
analizer=None,
temp_dir=directory
)
......@@ -178,7 +178,7 @@ def test_lightcnn_trainer():
shutil.rmtree(directory)
del trainer
del graph
"""
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment