mjc92 / WideResNet-pytorch

Wide Residual Networks (WideResNets) in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

Wide Residual Networks (WideResNets) in PyTorch


Languages

Language:Python 100.0%