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

distilbert: fix number of hidden_size

parent 0ecfd17f
......@@ -95,7 +95,7 @@ class DistilBertConfig(PretrainedConfig):
" or the path to a pretrained model config file (str)")
@property
def hidden_size(self):
return self.hidden_dim
return self.dim
@property
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