https://github.com/eurus-holmes/deit-cifar
https://github.com/eurus-holmes/deit-cifar
Last synced: 3 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/eurus-holmes/deit-cifar
- Owner: Eurus-Holmes
- Created: 2022-02-26T03:21:24.000Z (about 3 years ago)
- Default Branch: main
- Last Pushed: 2022-03-02T23:51:25.000Z (about 3 years ago)
- Last Synced: 2025-02-27T00:54:30.163Z (3 months ago)
- Language: Jupyter Notebook
- Size: 36.1 KB
- Stars: 1
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- Dei: DeiT_CIFAR.ipynb
Awesome Lists containing this project
README
# Finetune Deit on CIFAR-10
The code was adapted from [DeiT](https://github.com/facebookresearch/deit), but make sure you have the [pytorch-image-models](https://github.com/rwightman/pytorch-image-models) package `timm==0.4.12`.
```
pip install timm==0.4.12
```
## Example
With DeiT-tiny:
```
python main.py --finetune https://dl.fbaipublicfiles.com/deit/deit_tiny_patch16_224-a1311bcf.pth --data-set CIFAR-10 --model deit_tiny_patch16_224 --batch-size 128 --epochs 20 --output_dir tiny_CIFAR10
```
With DeiT-base:
```
python main.py --finetune https://dl.fbaipublicfiles.com/deit/deit_base_patch16_224-b5f2ef4d.pth --data-set CIFAR-10 --model deit_base_patch16_224 --batch-size 128 --epochs 20 --output_dir output
```