(self, labels, train, force_drop_ids=None)
| 119 | return labels |
| 120 | |
| 121 | def forward(self, labels, train, force_drop_ids=None): |
| 122 | use_dropout = self.dropout_prob > 0 |
| 123 | if (train and use_dropout) or (force_drop_ids is not None): |
| 124 | labels = self.token_drop(labels, force_drop_ids) |
| 125 | embeddings = self.embedding_table(labels) |
| 126 | return embeddings |
| 127 | |
| 128 | |
| 129 | ############################################################################# |
nothing calls this directly
no test coverage detected