Requirements

See the installation instruction for a step-by-step installation guide. See the server instruction for server settup.

Install cuda-8.0
Install cudnn v5.1
Download Pytorch 2.7 and clone the repository.

pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp27-none-linux_x86_64.whl
pip install torchvision
git clone https://github.com/meliketoy/wide-resnet.pytorch

How to run

After you have cloned the repository, you can train each dataset of either cifar10, cifar100 by running the script below.

python main --lr 0.1 resume false --net_type [lenet/vggnet/resnet/wide-resnet] --depth 28 --widen_factor 10 --dropout_rate 0.3 --dataset [cifar10/cifar100]

Implementation Details

epoch	learning rate	weight decay	Optimizer	Momentum	Nesterov
0 ~ 60	0.1	0.0005	Momentum	0.9	true
61 ~ 120	0.02	0.0005	Momentum	0.9	true
121 ~ 160	0.004	0.0005	Momentum	0.9	true
161 ~ 200	0.0008	0.0005	Momentum	0.9	true

CIFAR-10 Results

Below is the result of the test set accuracy for CIFAR-10 dataset training.

Accuracy is the average of 5 runs

network	dropout	preprocess	GPU:0	GPU:1	per epoch	accuracy(%)
wide-resnet 28x10	0	ZCA	5.90G	-	2 min 03 sec	95.83
wide-resnet 28x10	0	meanstd	5.90G	-	2 min 03 sec	96.21
wide-resnet 28x10	0.3	meanstd	5.90G	-	2 min 03 sec	96.27
wide-resnet 28x20	0.3	meanstd	8.13G	6.93G	4 min 10 sec	96.55
wide-resnet 40x10	0.3	meanstd	8.08G	-	3 min 13 sec	96.31
wide-resnet 40x14	0.3	meanstd	7.37G	6.46G	3 min 23 sec	96.34

CIFAR-100 Results

Below is the result of the test set accuracy for CIFAR-100 dataset training.

Accuracy is the average of 5 runs

network	dropout	preprocess	GPU:0	GPU:1	per epoch	Top1 acc(%)	Top5 acc(%)
wide-resnet 28x10	0	ZCA	5.90G	-	2 min 03 sec	80.07	95.02
wide-resnet 28x10	0	meanstd	5.90G	-	2 min 03 sec	81.02	95.41
wide-resnet 28x10	0.3	meanstd	5.90G	-	2 min 03 sec	81.49	95.62
wide-resnet 28x20	0.3	meanstd	8.13G	6.93G	4 min 05 sec	82.45	96.11
wide-resnet 40x10	0.3	meanstd	8.93G	-	3 min 06 sec	81.42	95.63
wide-resnet 40x14	0.3	meanstd	7.39G	6.46G	3 min 23 sec	81.87	95.51

License

bmsookim/wide-resnet.pytorch

ErrorLooks like something went wrong!

About

Resources

License

Uh oh!

Stars

Watchers

Forks

Releases

Packages 0

Uh oh!

Languages

Error
Looks like something went wrong!

Packages