| 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 |