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
Branches
Tags
1 merge request!93ArcFace/SphereFace Loss
Pipeline #47383 passed
...@@ -287,7 +287,7 @@ def add_bottleneck(model, bottleneck_size=128, dropout_rate=0.2): ...@@ -287,7 +287,7 @@ def add_bottleneck(model, bottleneck_size=128, dropout_rate=0.2):
new_model.add(GlobalAvgPool2D()) new_model.add(GlobalAvgPool2D())
new_model.add(Dropout(dropout_rate, name="Dropout")) 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")) new_model.add(BatchNormalization(axis=-1, scale=False, name="embeddings/BatchNorm"))
return new_model return new_model
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment