Method
__init__
(self,
attn_cfgs,
feedforward_channels,
ffn_dropout=0.0,
operation_order=None,
act_cfg=dict(type='ReLU', inplace=True),
norm_cfg=dict(type='LN'),
ffn_num_fcs=2,
**kwargs)
Source from the content-addressed store, hash-verified
| 310 | @TRANSFORMER_LAYER.register_module() |
| 311 | class LATRDecoderLayer(BaseTransformerLayer): |
| 312 | def __init__(self, |
| 313 | attn_cfgs, |
| 314 | feedforward_channels, |
| 315 | ffn_dropout=0.0, |
| 316 | operation_order=None, |
| 317 | act_cfg=dict(type='ReLU', inplace=True), |
| 318 | norm_cfg=dict(type='LN'), |
| 319 | ffn_num_fcs=2, |
| 320 | **kwargs): |
| 321 | super(LATRDecoderLayer, self).__init__( |
| 322 | attn_cfgs=attn_cfgs, |
| 323 | feedforward_channels=feedforward_channels, |
| 324 | ffn_dropout=ffn_dropout, |
| 325 | operation_order=operation_order, |
| 326 | act_cfg=act_cfg, |
| 327 | norm_cfg=norm_cfg, |
| 328 | ffn_num_fcs=ffn_num_fcs, |
| 329 | **kwargs) |
| 330 | |
| 331 | def forward(self, |
| 332 | query, |
Callers
nothing calls this directly
Tested by
no test coverage detected