Well one quick and dirty hack would be for your CustomDataset to return a very high number (e.g. np.iinfo(np.int64).max) in its __len__ .
As for get_next(), you can get the iterator from the dataloader and call next on that:
next(dataloader.__iter__())