Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch_transformers/modeling_distilbert.py
View file @ 0f8ad892
...@@ -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