(self, data, seq_len)
| 54 | |
| 55 | class TextSamplerDataset(Dataset): |
| 56 | def __init__(self, data, seq_len): |
| 57 | super().__init__() |
| 58 | self.data = data |
| 59 | self.seq_len = seq_len |
| 60 | |
| 61 | def __getitem__(self, index): |
| 62 | rand_start = torch.randint(0, self.data.size(0) - self.seq_len, (1,)) |
nothing calls this directly
no outgoing calls
no test coverage detected