(self, labels, train, force_drop_ids=None)
| 303 | return labels |
| 304 | |
| 305 | def forward(self, labels, train, force_drop_ids=None): |
| 306 | use_dropout = self.dropout_prob > 0 |
| 307 | if (train and use_dropout) or (force_drop_ids is not None): |
| 308 | labels = self.token_drop(labels, force_drop_ids) |
| 309 | embeddings = self.embedding_table(labels) |
| 310 | return embeddings |
| 311 | |
| 312 | |
| 313 | class FinalLayer(nn.Module): |
nothing calls this directly
no test coverage detected