Collate function is not called in dataloader

data
2

This seems to be expected, since the PyTorch-Geometric DataLoader implementation will delete your custom collate_fn and replace it with their Collater class as seen here.