├── LICENSE ├── README.md ├── data.py ├── main_gbn.py ├── main_normal.py ├── models ├── __init__.py ├── alexnet.py ├── cifar_shallow.py ├── mnist_f1.py ├── resnet.py └── wide_resnet.py ├── preprocess.py ├── requirements.txt ├── run_experiments.sh └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/README.md -------------------------------------------------------------------------------- /data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/data.py -------------------------------------------------------------------------------- /main_gbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/main_gbn.py -------------------------------------------------------------------------------- /main_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/main_normal.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/models/alexnet.py -------------------------------------------------------------------------------- /models/cifar_shallow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/models/cifar_shallow.py -------------------------------------------------------------------------------- /models/mnist_f1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/models/mnist_f1.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/wide_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/models/wide_resnet.py -------------------------------------------------------------------------------- /preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/preprocess.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_experiments.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/run_experiments.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/eladhoffer/bigBatch/HEAD/utils.py --------------------------------------------------------------------------------