Upload vit_best_model_optuna.pth
Browse filesHYPER PARAMETER TUNING -
We achieved a low accuracy level so to tune it and refine it we decided to tune the hyper parameters such as value, batch_size, learning_rate and weight_decay
Best hyperparameters found:
{'lr': 1.0369533463569177e-05, 'batch_size': 32, 'weight_decay': 0.009006790996778456}
Accuracy: 77.82%
Model saved at: vit_best_model_optuna.pth

vit_best_model_optuna.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4b51d4c16d25137d2b1a43532b264dbe59553b81b429055ab5c028b48d6e9d4a
|
| 3 |
+
size 343259379
|