(self, config, add_pooling_layer=True)
| 654 | """ |
| 655 | |
| 656 | def __init__(self, config, add_pooling_layer=True): |
| 657 | super().__init__(config) |
| 658 | self.config = config |
| 659 | |
| 660 | self.embeddings = BertEmbeddings(config) |
| 661 | |
| 662 | self.encoder = BertEncoder(config) |
| 663 | |
| 664 | self.pooler = BertPooler(config) if add_pooling_layer else None |
| 665 | |
| 666 | self.init_weights() |
| 667 | |
| 668 | |
| 669 | def get_input_embeddings(self): |
nothing calls this directly
no test coverage detected