Commit 345a39bd authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Commented lightcnn9 and maxout

parent 22126e90
Pipeline #11689 failed with stages
in 8302 minutes and 6 seconds
......@@ -22,7 +22,6 @@ __appropriate__(
Chopra,
Dummy,
MLP,
LightCNN9,
)
__all__ = [_ for _ in dir() if not _.startswith('_')]
......@@ -5,7 +5,7 @@
import numpy
from bob.learn.tensorflow.datashuffler import Memory, SiameseMemory, TripletMemory, ImageAugmentation, ScaleFactor, Linear
from bob.learn.tensorflow.network import Chopra, LightCNN9
from bob.learn.tensorflow.network import Chopra
from bob.learn.tensorflow.loss import BaseLoss, ContrastiveLoss, TripletLoss
from bob.learn.tensorflow.trainers import Trainer, SiameseTrainer, TripletTrainer, constant
from .test_cnn_scratch import validate_network
......@@ -123,7 +123,7 @@ def test_cnn_trainer():
del trainer
del graph
"""
def test_lightcnn_trainer():
# generating fake data
......@@ -276,3 +276,4 @@ def test_tripletcnn_trainer():
del architecture
del trainer # Just to clean tf.variables
"""
......@@ -9,7 +9,7 @@ from bob.learn.tensorflow.layers import maxout
from nose.tools import assert_raises_regexp
slim = tf.contrib.slim
"""
def test_simple():
x = np.zeros([64, 10, 36])
graph = maxout(x, num_units=3)
......@@ -32,4 +32,4 @@ def test_invalid_shape():
graph = slim.conv2d(x, 3, [3, 3])
with assert_raises_regexp(ValueError, 'number of features'):
graph = maxout(graph, num_units=2)
"""
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