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

Fixed issue with the bottleneck_size

parent babec5d4
No related branches found
No related tags found
1 merge request!93ArcFace/SphereFace Loss
Pipeline #47383 passed
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment