---
base_model: google/vit-base-patch16-224-in21k
datasets:
- garythung/trashnet
license: apache-2.0
metrics:
- recall
- precision
- accuracy
model-index:
- name: vit-trash
  results: []
---

# vit-trash

This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an [garythung/trashnet](https://huggingface.co/datasets/garythung/trashnet) dataset.

## Training and evaluation data

```json
DatasetDict({
    train: Dataset({
        features: ['image', 'label'],
        num_rows: 3537
    })
    validation: Dataset({
        features: ['image', 'label'],
        num_rows: 758
    })
    test: Dataset({
        features: ['image', 'label'],
        num_rows: 759
    })
})
```
### Training hyperparameters
```json
TrainingArguments(
    output_dir="output-models",
    save_total_limit=2,
    report_to="wandb",
    save_strategy="epoch",
    evaluation_strategy="epoch",
    learning_rate=2e-5,
    per_device_train_batch_size=15,
    per_device_eval_batch_size=10,
    num_train_epochs=3,
    weight_decay=0.01,
    load_best_model_at_end=True,
    logging_dir="logs",
    remove_unused_columns=False,
)
```
### Training results



### Framework versions

- Transformers 4.41.0
- TensorFlow 2.15.0
- Datasets 2.19.1
- Tokenizers 0.19.1