Skip to content

bob.learn.tensorflow
bob.learn.tensorflow
Commit ce8740dc authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files
Options

Fixed issue with the bottleneck_size

parent babec5d4
Pipeline #47383 passed with stage
    in 3 minutes and 39 seconds
    Hide whitespace changes
    Inline Side-by-side
    Showing with 1 addition and 1 deletion
    bob/learn/tensorflow/layers.py
    View file @ ce8740dc
    ......@@ -287,7 +287,7 @@ def add_bottleneck(model, bottleneck_size=128, dropout_rate=0.2):
    new_model.add(GlobalAvgPool2D())
    new_model.add(Dropout(dropout_rate, name="Dropout"))
    new_model.add(Dense(128, use_bias=False, name="embeddings"))
    new_model.add(Dense(bottleneck_size, use_bias=False, name="embeddings"))
    new_model.add(BatchNormalization(axis=-1, scale=False, name="embeddings/BatchNorm"))
    return new_model
    ......
    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