File size: 765 Bytes
b94742f e5968a2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | ---
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/blob/main/compute_acc.py)
|