About ImageNet benchmark

[moskomule]

Sorry, my code was wrong, and after fixing it, it works as expected.

---

Hi, I recently found this project and am interested in it.
In particular, the benchmark on ImageNet is amazing.

However, at the moment, I can only find its simplified example https://pytorch.org/tensordict/tutorials/tensorclass_imagenet.html.
I'm trying to modify it for ImageNet, then data.memmap_() in ImageNetData.from_dataset seems to take a very long time, even for the smaller validation dataset.

Can you please share the code to reproduce the experiment or let me know the tricks to use this example for larger problems?

Thank you!