| library_name: pytorch | |
| pipeline_tag: image-classification | |
| datasets: | |
| - food101 | |
| tags: | |
| - food101 | |
| - image-classification | |
| - vision-transformer | |
| - vit | |
| license: mit | |
| # ViT-B-16 on Food-101 | |
| Raw training checkpoint exported from this repository: | |
| https://github.com/mnjm/vit | |
| ## Files | |
| - `ViT-B-16-food101.pt`: raw checkpoint from this training run | |
| - `config/`: Hydra config snapshot saved with the run | |
| ## Checkpoint | |
| - Path: `ViT-B-16-food101.pt` | |
| - Dataset: `food101` | |
| - Model: `ViT-B/16` | |
| ## Metrics | |
| | Split | Loss | Acc@1 | Acc@5 | | |
| | --- | ---: | ---: | ---: | | |
| | Train | `0.0227` | `99.41%` | `99.87%` | | |
| | Val | `2.5183` | `58.57%` | `81.41%` | | |
| ## Usage | |
| This is a raw checkpoint dump. To use it check [here](https://github.com/mnjm/vit/compute_acc.py) | |