Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Nitin Kumar Bansal authored Oct 23, 2018
1 parent cf4cf7d commit 5204002
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,9 @@ Figure 1. Validation Curve Achieved for differnet Regularizers Proposed
- [x] Mutual Coherence Based
- [x] Restricted Isometry (**Best Performing** )

## Usage Wide-Resnet CIFAR
### Usage Wide-Resnet CIFAR
For CIFAR datasets,we choose Wide Resnet Arch. with a depth of 28 and Kernel width of 10,which
gives the best results for comparable number parameters for any Wide-Resnet Model.
To train on Cifar-10 using 2 gpu:

```bash
Expand All @@ -46,10 +48,13 @@ CUDA_VISIBLE_DEVICES=6,7 python train_n.py --ngpu 2 --dataset cifar100
After train phase, you can check saved model in the ```runs``` folder.

## Usage Wide-Resnet SVHN
For SVHN datasets,we choose Wide Resnet Arch. with a depth of 16 and Kernel width of 8,which
gives the best results for comparable number parameters for any Wide-Resnet Model.
``` bash
CUDA_VISIBEL_DEVICES=0 python train.py --dataset svhn --model wideresnet --learning_rate 0.01 --epochs 160
```

## Result
| **Network** | **CIFAR-10** | **CIFAR-100** | **SVHN** |
| ----------- | ------------ | ------------- | -------- |
| WideResNet | 4.16 | 20.50 | 1.60 |
Expand Down

0 comments on commit 5204002

Please sign in to comment.