diff --git a/example/README.md b/example/README.md index 49c92d2..2afc937 100644 --- a/example/README.md +++ b/example/README.md @@ -2,3 +2,5 @@ This folder contains a simple Wide-ResNet implementation that can be trained on Cifar10/SVHN with SAM. Start the training by running `python3 train-(cifar/svhn).py` + +SVHN data loader is taken from: https://github.com/uoguelph-mlrg/Cutout