Update README.md
Browse files
README.md
CHANGED
@@ -60,8 +60,14 @@ The base model used is the Vision Transformer `vit-base-patch16-224-in21k`, whic
|
|
60 |
The model was trained on a curated dataset of UAE company logos as well as others of international companies. The dataset consists of thousands of images across various brands to ensure robustness and accuracy.
|
61 |
|
62 |
### Performance
|
63 |
-
|
64 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
65 |
## How to Use
|
66 |
To use the model for inference, you can load it using the `transformers` library from Hugging Face:
|
67 |
|
@@ -154,7 +160,7 @@ print(model.config.id2label[predicted_label])
|
|
154 |
|
155 |
|
156 |
### Limitations and Biases
|
157 |
-
- The model is specifically trained on UAE company logos and may not perform well on logos from
|
158 |
- The model's performance is contingent upon the quality and diversity of the training dataset.
|
159 |
- Potential biases in the training data can lead to biases in model predictions.
|
160 |
|
|
|
60 |
The model was trained on a curated dataset of UAE company logos as well as others of international companies. The dataset consists of thousands of images across various brands to ensure robustness and accuracy.
|
61 |
|
62 |
### Performance
|
63 |
+
***** eval metrics *****
|
64 |
+
- epoch = 20.0
|
65 |
+
- eval_accuracy = 0.9761
|
66 |
+
- eval_loss = 0.1193
|
67 |
+
- eval_runtime = 0:00:20.51
|
68 |
+
- eval_samples_per_second = 268.951
|
69 |
+
- eval_steps_per_second = 8.432
|
70 |
+
|
71 |
## How to Use
|
72 |
To use the model for inference, you can load it using the `transformers` library from Hugging Face:
|
73 |
|
|
|
160 |
|
161 |
|
162 |
### Limitations and Biases
|
163 |
+
- The model is specifically trained on UAE company logos and a few from Outside the UAE, it may not perform well on logos from numerous other large or small companies.
|
164 |
- The model's performance is contingent upon the quality and diversity of the training dataset.
|
165 |
- Potential biases in the training data can lead to biases in model predictions.
|
166 |
|