| 95 | |
| 96 | |
| 97 | class FakeDataset(TensorDataset): |
| 98 | def __init__(self, process_fn=None, *args, **kwargs): |
| 99 | super().__init__(*args, **kwargs) |
| 100 | self.process_fn = process_fn |
| 101 | |
| 102 | def __getitem__(self, idx): |
| 103 | return self.process_fn(super().__getitem__(idx)) |
| 104 | |
| 105 | def create_dataset_function(path, args): |
| 106 | num_samples = 100000 |
no outgoing calls