(dataset, batch_size=2, num_workers=3, shuffle=True, collate_fn=custom_collate_fn) assert dataset._datasets[0].shuffle assert dataset._datasets[1].shuffle dataloader_iter = iter(dataloader) assert next(dataloader_iter) == "received" assert dataloader._num_samples_yielded_combined[0] == [2...
🐛 Bug I wrote a custom collect_fn to filter out bad samples. But in some batches, all samples are being removed so it returns an empty list. But in the default_collate(batch) fn, it assumes the batch is not None since it immediately read...