prashanth0205
/

vit_spectrogram

Image Classification

generated_from_keras_callback

Model card Files Files and versions

prashanth0205 commited on Jul 8, 2022

Commit

8555f5d

·

1 Parent(s): 9139d20

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -12,7 +12,8 @@ probably proofread and complete it, then remove this comment. -->
 # vit_spectrogram
-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 unknown dataset.
 It achieves the following results on the evaluation set:
 - Train Loss: 0.2893
 - Train Accuracy: 0.8757

 # vit_spectrogram
+This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on a dataset containing
+images of Mel spectrogram belonging to the classes 'Male' and 'Female'. This model is still being fine tuned and tested.
 It achieves the following results on the evaluation set:
 - Train Loss: 0.2893
 - Train Accuracy: 0.8757