Update README.md

README.md

@@ -56,7 +56,7 @@ Batch size: 32 for both training and evaluation
 Seed: 42 (for reproducibility)
 Optimizer: Adam (with betas=(0.9, 0.999) and epsilon=1e-08)
 Scheduler: Linear learning rate scheduler
-Number of epochs: 2
+Number of epochs: 5
 The training used a combination of cross-entropy loss for multi-class classification and early stopping based on evaluation metrics.
 
 ### Training hyperparameters
@@ -68,7 +68,7 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 2
+- num_epochs: 5
 
 ### Framework versions