Skip to content

Commit 3a795a0

Browse files
authored
Doc: Fix parameter name typo in BertBackbone docstring (keras-team#2529)
* Doc: Fix parameter name typo in BertBackbone docstring * Style: Apply Black and Isort formatting * Style: Fix linting errors with Ruff
1 parent b98945d commit 3a795a0

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

keras_hub/src/models/bert/bert_backbone.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ class BertBackbone(Backbone):
3333
vocabulary_size: int. The size of the token vocabulary.
3434
num_layers: int. The number of transformer layers.
3535
num_heads: int. The number of attention heads for each transformer.
36-
The hidden size must be divisible by the number of attention heads.
36+
The hidden_dim must be divisible by the number of attention heads.
3737
hidden_dim: int. The size of the transformer encoding and pooler layers.
3838
intermediate_dim: int. The output dimension of the first Dense layer in
3939
a two-layer feedforward network for each transformer.

0 commit comments

Comments
 (0)