Commit 9d8c3b8e authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Added test case LightCNN9

parent 5f92d884
Pipeline #11661 failed with stages
in 6 minutes and 18 seconds
......@@ -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
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment