Dataset download faster

from datasets import load_dataset
dataset = load_dataset(“imagenet-1k”, split = “train”)

I have seen that the above code downloads the data in partitions .
How can I just download the first partition and not waste any more time , since I donot need whole data . I just need some data for analysis . But , I have to wait for the whole time

Of course, you can use the load_size parameter to quickly load only part of the data. This option allows you to specify the number of elements you want to load from the dataset. For example, you can set load_size=100 to load only the first 100 elements. This way, you can quickly get a small set of data for analysis without downloading the entire database. This saves time and resources when working with large data sets.