(self, config, add_pooling_layer=True)
| 883 | """ |
| 884 | |
| 885 | def __init__(self, config, add_pooling_layer=True): |
| 886 | super().__init__(config) |
| 887 | self.config = config |
| 888 | |
| 889 | self.embeddings = BertEmbeddings(config) |
| 890 | self.encoder = BertEncoder(config) |
| 891 | |
| 892 | self.pooler = BertPooler(config) if add_pooling_layer else None |
| 893 | |
| 894 | # Initialize weights and apply final processing |
| 895 | self.post_init() |
| 896 | |
| 897 | def get_input_embeddings(self): |
| 898 | return self.embeddings.word_embeddings |
nothing calls this directly
no test coverage detected