Skip to content

Commit

Permalink
F
Browse files Browse the repository at this point in the history
ixed missing separator between layers
  • Loading branch information
muellerdo authored May 28, 2024
1 parent 77f9774 commit 7b9c7da
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion vit_keras/layers.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,7 +143,7 @@ def build(self, input_shape):
),
tf.keras.layers.Lambda(
lambda x: tf.keras.activations.gelu(x, approximate=False)
)
),
tf.keras.layers.Dropout(self.dropout),
tf.keras.layers.Dense(input_shape[-1], name=f"{self.name}/Dense_1"),
tf.keras.layers.Dropout(self.dropout),
Expand Down

0 comments on commit 7b9c7da

Please sign in to comment.