(self,
*args,
post_norm_cfg=dict(type='LN'),
return_intermediate=False,
**kwargs)
| 142 | `LN`. |
| 143 | """ |
| 144 | def __init__(self, |
| 145 | *args, |
| 146 | post_norm_cfg=dict(type='LN'), |
| 147 | return_intermediate=False, |
| 148 | **kwargs): |
| 149 | |
| 150 | super(DetrTransformerDecoder, self).__init__(*args, **kwargs) |
| 151 | self.return_intermediate = return_intermediate |
| 152 | if post_norm_cfg is not None: |
| 153 | self.post_norm = build_norm_layer(post_norm_cfg, |
| 154 | self.embed_dims)[1] |
| 155 | else: |
| 156 | self.post_norm = None |
| 157 | |
| 158 | def forward(self, query, *args, **kwargs): |
| 159 | """Forward function for `TransformerDecoder`. |