(self, config)
| 14 | # self.pooler_output = pooler_output |
| 15 | |
| 16 | def __init__(self, config): |
| 17 | RobertaForSequenceClassification.__init__(self, config) |
| 18 | self.embeddingHead = nn.Linear(config.hidden_size, 768) |
| 19 | self.norm = nn.LayerNorm(768) |
| 20 | self.apply(self._init_weights) |
| 21 | self.use_mean = False |
| 22 | |
| 23 | def _init_weights(self, module): |
| 24 | """ Initialize the weights """ |
nothing calls this directly
no outgoing calls
no test coverage detected