├── .gitignore ├── LICENSE ├── README.md ├── configs.txt ├── data.py ├── evaluate.py ├── get_sharpnesses ├── main.py ├── main_gbn.py ├── main_normal.py ├── measure_sharpness.py ├── models ├── __init__.py ├── alexnet.py ├── alexnetv2.py ├── alexnetv3.py ├── alexnetvlr.py ├── cifar_shallow.py ├── mnist_f1.py ├── noisy_relu.py ├── resnet.py └── wide_resnet.py ├── preprocess.py ├── requirements.txt ├── run_experiments.sh ├── test.py ├── utils.py ├── visualize2_sharpness.py └── visualize_sharpness.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | TrainingResults 3 | pytorch-gen 4 | .idea 5 | *.pdf 6 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/README.md -------------------------------------------------------------------------------- /configs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/configs.txt -------------------------------------------------------------------------------- /data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/data.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/evaluate.py -------------------------------------------------------------------------------- /get_sharpnesses: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/get_sharpnesses -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/main.py -------------------------------------------------------------------------------- /main_gbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/main_gbn.py -------------------------------------------------------------------------------- /main_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/main_normal.py -------------------------------------------------------------------------------- /measure_sharpness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/measure_sharpness.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/models/alexnet.py -------------------------------------------------------------------------------- /models/alexnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/models/alexnetv2.py -------------------------------------------------------------------------------- /models/alexnetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/models/alexnetv3.py -------------------------------------------------------------------------------- /models/alexnetvlr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/models/alexnetvlr.py -------------------------------------------------------------------------------- /models/cifar_shallow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/models/cifar_shallow.py -------------------------------------------------------------------------------- /models/mnist_f1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/models/mnist_f1.py -------------------------------------------------------------------------------- /models/noisy_relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/models/noisy_relu.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/wide_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/models/wide_resnet.py -------------------------------------------------------------------------------- /preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/preprocess.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_experiments.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/run_experiments.sh -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/test.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/utils.py -------------------------------------------------------------------------------- /visualize2_sharpness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/visualize2_sharpness.py -------------------------------------------------------------------------------- /visualize_sharpness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenwei202/smoothout/HEAD/visualize_sharpness.py --------------------------------------------------------------------------------