Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
bob
bob.learn.tensorflow
Commits
ce8740dc
Commit
ce8740dc
authored
Jan 26, 2021
by
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
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
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
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment