mirrors/mmcv
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmcv.git synced 2025-06-03 21:54:52 +08:00
Code Issues Projects Releases Wiki Activity

Remove redundant self.init_cfg (#959)

Browse Source
This commit is contained in:
ZhangShilong 2021-04-17 10:06:44 +08:00 committed by GitHub
parent ae411509dd
commit cb0ee1e339
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmcv/cnn/bricks/transformer.py
View File

@ -61,7 +61,6 @@ class MultiheadAttention(BaseModule):
self.attn = nn.MultiheadAttention(embed_dims, num_heads, dropout,
**kwargs)
self.dropout = nn.Dropout(dropout)
self.init_cfg = init_cfg
def forward(self,
query,
@ -171,7 +170,6 @@ class FFN(BaseModule):
self.num_fcs = num_fcs
self.act_cfg = act_cfg
self.dropout = dropout
self.init_cfg = init_cfg
self.activate = build_activation_layer(act_cfg)
layers = []
@ -422,7 +420,6 @@ class TransformerLayerSequence(BaseModule):
else:
assert isinstance(transformerlayers, list) and \
len(transformerlayers) == num_layers
self.init_cfg = init_cfg
self.num_layers = num_layers
operation_order = transformerlayers[0]['operation_order']
self.pre_norm = operation_order[0] == 'norm'