()
| 39 | criterion = torch.nn.CrossEntropyLoss() |
| 40 | |
| 41 | def get_next_batch(): |
| 42 | for _ in range(5): |
| 43 | data = torch.LongTensor(batch, nindices) % ntoken |
| 44 | target = torch.LongTensor(batch, ntoken) % nindices |
| 45 | yield data, target |
| 46 | |
| 47 | # train for 10 iterations |
| 48 | for epoch in range(10): |