Heem2's picture
Update README.md
86aa98c verified
metadata
license: apache-2.0
base_model: google/vit-base-patch16-224-in21k
tags:
  - generated_from_trainer
metrics:
  - accuracy
model-index:
  - name: New-plant-diseases-classification
    results: []

New-plant-diseases-classification

This model is a fine-tuned version of google/vit-base-patch16-224-in21k on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0548
  • Accuracy: 0.995

Training results

Training Loss Epoch Step Validation Loss Accuracy
2.5635 1.0 137 1.0712 0.9918
0.5929 2.0 275 0.3213 0.9923
0.2239 3.0 412 0.1482 0.9955
0.1256 4.0 550 0.1175 0.9882
0.0807 5.0 687 0.0648 0.9955
0.0561 6.0 825 0.0548 0.995

Framework versions

  • Transformers 4.38.2
  • Pytorch 2.1.2
  • Datasets 2.18.0
  • Tokenizers 0.15.2