| 2 | from torch.utils.data import Dataset |
| 3 | |
| 4 | class MyTrainDataset(Dataset): |
| 5 | def __init__(self, size): |
| 6 | self.size = size |
| 7 | self.data = [(torch.rand(20), torch.rand(1)) for _ in range(size)] |
| 8 | |
| 9 | def __len__(self): |
| 10 | return self.size |
| 11 | |
| 12 | def __getitem__(self, index): |
| 13 | return self.data[index] |
no outgoing calls
no test coverage detected