Table 5.
List of Hypermaters, loss and optimizer used for training classification and segmentation models.
| Parameters | Segmentation model | Classification model |
|---|---|---|
| Batch size | 32 | 64 |
| Learning Rate (Initial) | 0.0001 | 0.0001 |
| Epochs | 45 | 50 |
| Shuffle Each Iteration | Yes | Yes |
| Stopping criteria | 5 | 10 |
| Loss function | Negative Log Likelihood Loss/MSE | CELoss |
| Optimizer | SGDM | SGDM |