search

MadryLab / robustness

A library for experimenting with, training and evaluating neural networks, with a focus on adversarial robustness.
MIT License
903 stars 181 forks source link

imagenet dataset download #95

Closed GaoJiqiang closed 3 years ago

GaoJiqiang commented 3 years ago

I want to train a standard model on imagenet, but there has a problem when downloading the dataset.

image

Hadisalman commented 3 years ago

Hi @GaoJiqiang, you have to first download ImageNet before running the code (the code won't download it automatically for you).

Basically you have to obtain a copy of ImageNet (http://www.image-net.org/index), then preprocess the val directory to look like the train directory by running this script. Finally, point --data to the directory containing the train and val folders of ImageNet that you downloaded and processed.

Hope this helps!