Update README.md

README.md

@@ -18,7 +18,11 @@ model = AutoModelForImageClassification.from_pretrained("AEnigmista/Sardegna-ViT
 For more information on the code: please visit the [github repo](https://github.com/MatteoMocci/Most-Walkability-AI)
 # Training Hyper-parameters
 This version's hyper-parameters for training are:
-Fp16 = True, gradient accumulation steps = 4, batch size = 16, 20 epochs, learning rate = 2e-4
+- Fp16 = True
+- batch size = 32
+- 10 epochs
+- learning rate = 1e-4
+- optimizer = 'adamw_hf'
 
 # Metrics
 The metrics that are used for evaluation are accuracy, recall, precision, mse, confusion matrix and a custom metric called one_out. The one_out_accuracy uses