Custom Collete

[MLlab4CS]

How to pass custom collete function into PyG data loader?

[rusty1s]

The torch_geometric.loader.DataLoader makes use of the PyTorch DataLoader under the hood. If you want to override/customize the collate_fn, I suggest you make directly use of torch.utils.data.DataLoader.