Unverified Commit 564fd75d authored by guillaume-be's avatar guillaume-be Committed by GitHub
Browse files

Removed unused fields in DistilBert TransformerBlock (#2710)

* Removed unused fields in DistilBert TransformerBlock
parent 889d3bfd
...@@ -217,11 +217,6 @@ class TransformerBlock(nn.Module): ...@@ -217,11 +217,6 @@ class TransformerBlock(nn.Module):
def __init__(self, config): def __init__(self, config):
super().__init__() super().__init__()
self.n_heads = config.n_heads
self.dim = config.dim
self.hidden_dim = config.hidden_dim
self.dropout = nn.Dropout(p=config.dropout)
self.activation = config.activation
self.output_attentions = config.output_attentions self.output_attentions = config.output_attentions
assert config.dim % config.n_heads == 0 assert config.dim % config.n_heads == 0
......
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