From ce8740dc37c6c7648dcedf144fd2ddb25a2acc65 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Tue, 26 Jan 2021 17:11:36 +0100
Subject: [PATCH] Fixed issue with the bottleneck_size
---
bob/learn/tensorflow/layers.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/bob/learn/tensorflow/layers.py b/bob/learn/tensorflow/layers.py
index aa051071..6c2f4c06 100644
--- a/bob/learn/tensorflow/layers.py
+++ b/bob/learn/tensorflow/layers.py
@@ -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
--
GitLab