Skip to content

GitLab

Commit ed2ab1c2 authored by Stefan Schweter's avatar Stefan Schweter
Browse files

distilbert: fix number of hidden_size

parent 0ecfd17f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch_transformers/modeling_distilbert.py
View file @ ed2ab1c2
...@@ -95,7 +95,7 @@ class DistilBertConfig(PretrainedConfig): ...@@ -95,7 +95,7 @@ class DistilBertConfig(PretrainedConfig):
" or the path to a pretrained model config file (str)") " or the path to a pretrained model config file (str)")
@property @property
def hidden_size(self): def hidden_size(self):
return self.hidden_dim return self.dim
@property @property
def num_attention_heads(self): def num_attention_heads(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment