Downloads · 30 days
16
43% of all-time downloads
trash-classification-vit
suramadu08/trash-classification-vit
trash-classification-vit is a machine learning model from suramadu08. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
This model is a fine-tuned version of google/vit-base-patch16-224-in21k on an garythung/trashnet dataset.
All-time downloads
37
Public
Parameters
303M
2.4 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors1.2 GB · 100%
At a glance
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Oct 28, 2024
- Updated
- Oct 28, 2024
- SHA
- a2f4f9aa
Base models
Datasets
Model card
From the Hugging Face model README
vit-trash
This model is a fine-tuned version of google/vit-base-patch16-224-in21k on an garythung/trashnet dataset.
Training and evaluation data
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
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
- Type
- vit
- License
- apache-2.0
- Created
- Oct 28, 2024
- Updated
- Oct 28, 2024