https://github.com/hysts/pytorch_cutout
A PyTorch implementation of Cutout
https://github.com/hysts/pytorch_cutout
cifar10 computer-vision pytorch
Last synced: 9 months ago
JSON representation
A PyTorch implementation of Cutout
- Host: GitHub
- URL: https://github.com/hysts/pytorch_cutout
- Owner: hysts
- License: mit
- Created: 2017-12-23T05:22:21.000Z (over 8 years ago)
- Default Branch: master
- Last Pushed: 2018-06-02T01:18:09.000Z (about 8 years ago)
- Last Synced: 2025-10-06T18:43:35.351Z (9 months ago)
- Topics: cifar10, computer-vision, pytorch
- Language: Python
- Homepage:
- Size: 478 KB
- Stars: 16
- Watchers: 1
- Forks: 1
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# PyTorch Implementation of Cutout
## Usage
```
$ python main.py --block_type basic --depth 110 --use_cutout --cutout_size 16 --cutout_prob 1 --outdir results
```
## Results on CIFAR-10
| Model | Test Error (median of 5 runs) | Training Time |
|:----------------------------|:-----------------------------:|--------------:|
| ResNet-preact-56 w/o Cutout | 5.85 | 98 min |
| ResNet-preact-56 w/ Cutout | 4.96 | 98 min |
### w/o Cutout
```
$ python -u main.py --depth 56 --block_type basic --base_lr 0.2 --seed 7 --outdir results/wo_cutout/00
```
### w/ Cutout
```
$ python -u main.py --depth 56 --block_type basic --base_lr 0.2 --use_cutout --cutout_size 16 --cutout_prob 1 --seed 7 --outdir results/w_cutout/00
```
## References
* DeVries, Terrance, and Graham W. Taylor. "Improved regularization of convolutional neural networks with cutout." arXiv preprint arXiv:1708.04552 (2017). [arXiv:1708.04552]( https://arxiv.org/abs/1708.04552 ), [PyTorch implementation]( https://github.com/uoguelph-mlrg/Cutout )