(self, stage: str)
| 87 | |
| 88 | |
| 89 | def setup(self, stage: str): |
| 90 | data_by_split = get_era5_data_by_split(data_root=self.data_root, label_key=self.label_key, random_seed=0) |
| 91 | self.train_ds = TensorDataset(*data_by_split['train']) |
| 92 | self.valid_ds = TensorDataset(*data_by_split['val']) |
| 93 | self.evalu_ds = TensorDataset(*data_by_split['test']) |
| 94 | |
| 95 | self.test_locs = data_by_split['test'][0].detach() |
| 96 | |
| 97 | def train_dataloader(self): |
| 98 | return DataLoader(self.train_ds, batch_size=self.batch_size, num_workers=self.num_workers, shuffle=True) |
nothing calls this directly
no test coverage detected