| 110 | `LN`. Only used when `self.pre_norm` is `True` |
| 111 | """ |
| 112 | def __init__(self, *args, post_norm_cfg=dict(type='LN'), **kwargs): |
| 113 | super(DetrTransformerEncoder, self).__init__(*args, **kwargs) |
| 114 | if post_norm_cfg is not None: |
| 115 | self.post_norm = build_norm_layer( |
| 116 | post_norm_cfg, self.embed_dims)[1] if self.pre_norm else None |
| 117 | else: |
| 118 | assert not self.pre_norm, f'Use prenorm in ' \ |
| 119 | f'{self.__class__.__name__},' \ |
| 120 | f'Please specify post_norm_cfg' |
| 121 | self.post_norm = None |
| 122 | |
| 123 | def forward(self, *args, **kwargs): |
| 124 | """Forward function for `TransformerCoder`. |