Loading partial dataset when debugging

I am using PyTorch Lightning along with datasets (thanks for so many datasets already prepared and the great splits).

Every time I execute load_dataset for the imdb dataset it takes some time even if I specify a split involving very few samples. I guess this due to hashing as per the other issues.

Is there a way to only load part of the dataset on load_dataset? This would really speed up my workflow.
Something like a debug mode would really help. Thanks!