/WideResNet-pytorch

Wide Residual Networks (WideResNets) in PyTorch

Primary LanguagePython

Wide Residual Networks (WideResNets) in PyTorch

WideResNets for CIFAR10/100 implemented in PyTorch. This implementation requires less GPU memory than what is required by the official Torch implementation: https://github.com/szagoruyko/wide-residual-networks.

Example:

python train.py --dataset cifar100 --depth 40 --widen-factor 4

Tested on this version of PyTorch .

Acknowledgement