ffcv-imagenet icon indicating copy to clipboard operation
ffcv-imagenet copied to clipboard

numpy.core._exceptions._ArrayMemoryError: Unable to allocate 2.73 TiB for an array with shape (124994103978,) and data type [('sample_id', '<u8'), ('ptr', '<u8'), ('size', '<u8’)]

Open TheDarkKnight-21th opened this issue 11 months ago • 0 comments

i wanna train on imagenet21k, not imagenet1k. so i downloaded the imagenet21k(winter) on the official site.(imagenet official site)

and then i ran the "write_imagenet.sh" with default argument value (500 0.50 90) , setting the dataset imagenet21k(winter).

finally, i run the train_imagenet.py. and there is an error.

"numpy.core._exceptions._ArrayMemoryError: Unable to allocate 2.73 TiB for an array with shape (124994103978,) and data type [('sample_id', '<u8'), ('ptr', '<u8'), ('size', '<u8’)] "

what is that error? why does that error occur? (when i make a dataset with imagenet1k and run the train, there is not any error.) Is there any way to learn with imagenet21k?

(ref : The ffcv dataset that is made by "write_imagenet.sh" size is 2.73 TB.)

TheDarkKnight-21th avatar Feb 27 '24 16:02 TheDarkKnight-21th