Skip to content

GitLab

Unverified Commit cb0ee1e3 authored by ZhangShilong's avatar ZhangShilong Committed by GitHub
Browse files

Remove redundant self.init_cfg (#959)

parent ae411509
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
mmcv/cnn/bricks/transformer.py
View file @ cb0ee1e3
...@@ -61,7 +61,6 @@ class MultiheadAttention(BaseModule): ...@@ -61,7 +61,6 @@ class MultiheadAttention(BaseModule):
self.attn = nn.MultiheadAttention(embed_dims, num_heads, dropout, self.attn = nn.MultiheadAttention(embed_dims, num_heads, dropout,
**kwargs) **kwargs)
self.dropout = nn.Dropout(dropout) self.dropout = nn.Dropout(dropout)
self.init_cfg = init_cfg
def forward(self, def forward(self,
query, query,
...@@ -171,7 +170,6 @@ class FFN(BaseModule): ...@@ -171,7 +170,6 @@ class FFN(BaseModule):
self.num_fcs = num_fcs self.num_fcs = num_fcs
self.act_cfg = act_cfg self.act_cfg = act_cfg
self.dropout = dropout self.dropout = dropout
self.init_cfg = init_cfg
self.activate = build_activation_layer(act_cfg) self.activate = build_activation_layer(act_cfg)
layers = [] layers = []
...@@ -422,7 +420,6 @@ class TransformerLayerSequence(BaseModule): ...@@ -422,7 +420,6 @@ class TransformerLayerSequence(BaseModule):
else: else:
assert isinstance(transformerlayers, list) and \ assert isinstance(transformerlayers, list) and \
len(transformerlayers) == num_layers len(transformerlayers) == num_layers
self.init_cfg = init_cfg
self.num_layers = num_layers self.num_layers = num_layers
operation_order = transformerlayers[0]['operation_order'] operation_order = transformerlayers[0]['operation_order']
self.pre_norm = operation_order[0] == 'norm' self.pre_norm = operation_order[0] == 'norm'
......
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