pranav2711 commited on
Commit
6140576
·
verified ·
1 Parent(s): 7840edc

Upload vit_best_model_optuna.pth

Browse files

HYPER 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
![Screenshot 2025-05-20 161829.png](https://cdn-uploads.huggingface.co/production/uploads/682c0e3721ad41ff15dc242a/azjZEk-e4lOwGpsXmZMjk.png)

Files changed (1) hide show
  1. vit_best_model_optuna.pth +3 -0
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