We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 465e518 commit 1c27bd6Copy full SHA for 1c27bd6
pvnet/data/datamodule.py
@@ -24,7 +24,7 @@ def __len__(self):
24
25
def __getitem__(self, idx):
26
return torch.load(self.sample_paths[idx])
27
-
+
28
29
def collate_fn(samples: list[NumpyBatch]):
30
"""Convert a list of NumpyBatch samples to a tensor batch"""
0 commit comments