Table 6 Values of critical hyperparameters in training the EfficientNet-B0 model.
Hyperparameter | Value |
|---|---|
Number of layers | 250 |
Learning rate | 1e−3 for 1–500 epochs |
1e−5 for 501–1000 epochs | |
Optimizer | Adam |
Batch size | 16 |
Loss function | Categorical cross-entropy |
Number of epochs | 1000 |