File size: 1,317 Bytes
1b40673 2d6cb00 17b8573 2d6cb00 17b8573 1b40673 2d6cb00 |
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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
---
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 |