(self, ntoken, ninp, dropout)
| 40 | Encoding layers of the RNNModel |
| 41 | """ |
| 42 | def __init__(self, ntoken, ninp, dropout): |
| 43 | super(EmbeddingTable, self).__init__() |
| 44 | self.drop = nn.Dropout(dropout) |
| 45 | self.encoder = nn.Embedding(ntoken, ninp) |
| 46 | if torch.accelerator.is_available(): |
| 47 | device = torch.accelerator.current_accelerator() |
| 48 | self.encoder = self.encoder.to(device) |
| 49 | nn.init.uniform_(self.encoder.weight, -0.1, 0.1) |
| 50 | |
| 51 | def forward(self, input): |
| 52 | if torch.accelerator.is_available(): |