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

Fixed issue with the bottleneck_size

parent babec5d4
Pipeline #47383 passed with stage
in 3 minutes and 39 seconds
......@@ -287,7 +287,7 @@ def add_bottleneck(model, bottleneck_size=128, dropout_rate=0.2):
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
Supports Markdown
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