Method
__init__
(self, encoder=None, decoder=None, init_cfg=None)
Source from the content-addressed store, hash-verified
| 293 | Defaults to None. |
| 294 | """ |
| 295 | def __init__(self, encoder=None, decoder=None, init_cfg=None): |
| 296 | super(Transformer, self).__init__(init_cfg=init_cfg) |
| 297 | self.encoder = build_transformer_layer_sequence(encoder) |
| 298 | self.decoder = build_transformer_layer_sequence(decoder) |
| 299 | self.embed_dims = self.encoder.embed_dims |
| 300 | |
| 301 | def init_weights(self): |
| 302 | # follow the official DETR to init parameters |
Tested by
no test coverage detected