# Model Card for Model ID

This model is vit-base-patch16-224-in21k fine-tuned with a subset of the 2021 Kaggle Google Landmark Dataset competition, including only the top 51 categories. 
The dataset is available as Hugginface dataset on: https://huggingface.co/datasets/pemujo/GLDv2_Top_51_Categories


- **Developed by:** Pedro Melendez
- **Model type:** Vision transformer
- **Finetuned from model:** vit-base-patch16-224-in21k


### Training Data

Classes with more than 500 images in the 2021 Kaggle Google Landmark competition
https://huggingface.co/datasets/pemujo/GLDv2_Top_51_Categories

### Results

|                        |       |
|-----------------------:|------:|
| epoch                  |4      |
|eval_accuracy           |0.97411|
|eval_loss               |0.11560|
|eval_runtime (secs)     |79.0939|
|eval_samples_per_second |115.255|
|eval_steps_per_second   |14.413 |
|train_runtime (secs)    |4082.92|
|train_samples_per_second|35.722 |
|train_steps_per_second  |2.233  |


## Environmental Impact

- **Hardware Type:** Nvidia V100
- **Minutes used:** 68 Minutes
- **Cloud Provider:** Google Cloud
- **Compute Region:** us-central

### Compute Infrastructure

Google Cloud Workbench Instance

#### Hardware

GCP Workbench n1-highmem-8 instance with Nvidia V100 GPU

#### Software

Python 3.9
Pytorch 2.0.1+cu117
label_0 label_1 label_2 label_3 label_4 label_5 label_6 label_7 label_8 label_9 label_10 label_11 label_12 label_13 label_14 label_15 label_16 label_17 label_18 label_19 label_20 label_21 label_22 label_23 label_24 label_25 label_26 label_27 label_28 label_29 label_30 label_31 label_32 label_33 label_34 label_35 label_36 label_37 label_38 label_39 label_40 label_41 label_42 label_43 label_44 label_45 label_46 label_47 label_48 label_49 label_50