diff --git a/README.md b/README.md index 14f2363..28dc4b8 100644 --- a/README.md +++ b/README.md @@ -8,9 +8,17 @@ Our core message is that a very simple approach -- data augmentation with Gaussi ![Example Figure](./Figures/Fig1.png) +## Download models + +Download the trained model weights by running + +``` +bash ./Models/download_models.sh +``` + ## Evaluate models -To validate our models, run `python3 main.py` or run the provided bash script directly that downloads and evaluates all provided models via `bash run.sh`. +To validate our models, run `python3 main.py` or run the provided bash script that evaluates all provided models via `bash run.sh`. It is necessary to download the model weights first! Arguments: - `--datadir-clean` the top-level directory of the ImageNet dataset (mandatory)