(self, num_classes, hidden_size, dropout_prob)
| 281 | """ |
| 282 | |
| 283 | def __init__(self, num_classes, hidden_size, dropout_prob): |
| 284 | super().__init__() |
| 285 | use_cfg_embedding = dropout_prob > 0 |
| 286 | self.embedding_table = nn.Embedding( |
| 287 | num_classes + use_cfg_embedding, hidden_size |
| 288 | ) |
| 289 | self.num_classes = num_classes |
| 290 | self.dropout_prob = dropout_prob |
| 291 | |
| 292 | def token_drop(self, labels, force_drop_ids=None): |
| 293 | """ |