Table 6 Hyper parameters used for training
Parameters | Restnet-18 | MobileNet v2 | VGG19 |
|---|---|---|---|
Solver type | SGDM | SGDM | RMSProp |
Learning rate | 0.0001 | 0.0001 | 0.0001 |
Gradient threshold method | l2norm | l2norm | l2norm |
Loss function | Categorical cross-entropy | Categorical cross-entropy | Categorical cross-entropy |
L2 regularization | 0.01 | 0.01 | 0.01 |
Batch size | 64 | 64 | 64 |
Max epochs | 25 | 25 | 25 |
