Unverified Commit 0f8ad892 authored by Thomas Wolf's avatar Thomas Wolf Committed by GitHub
Browse files

Merge pull request #1135 from stefan-it/master

distilbert: fix number of hidden_size
parents 9ce42dc5 ed2ab1c2
......@@ -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