Adaout
Adaout is a practical and flexible regularization method with high generalization and interpretability.
Requirements
- python 3.6 (Anaconda version >=5.2.0 is recommended)
- torch (torch version >=1.1.0 is recommended)
- torchvision (torchvision version >=0.3.0 is recommended)
- pandas
- numpy
- NVIDIA GPU + CUDA CuDNN
Datasets
- CIFAR-10, CIFAR-100, SVHN, ImageNet and others
Getting started
- Download datasets and extract it inside
data
- Train:
python train.py
,python train100.py
orpython train_svhn.py
- Evaluate:
- Pretrained models for CIFAR-10 and CIFAR-100 are available at this link. Download and extract them in the
save_model/resnet56_10
orsave_model/resnet56_100
directory. - You should achieve about 94.63% accuracy on CIFAR-10, and 74.18% accuracy on CIFAR-100 datasets.
- Pretrained models for CIFAR-10 and CIFAR-100 are available at this link. Download and extract them in the