(self, num_samples=5000, batch_size=1000)
| 77 | |
| 78 | class LandOceanDataModule(pl.LightningDataModule): |
| 79 | def __init__(self, num_samples=5000, batch_size=1000): |
| 80 | super().__init__() |
| 81 | self.num_samples = num_samples |
| 82 | self.batch_size=batch_size |
| 83 | |
| 84 | def setup(self, stage: str): |
| 85 | self.train_ds = TensorDataset(*get_data(self.num_samples, seed=0)) |
nothing calls this directly
no outgoing calls
no test coverage detected