Table 4.
Optimal hyperparameters for pretrained model.
| Pretrained model | Hyperparameters | |||
|---|---|---|---|---|
| Split ratio | Optimizer | Learning rate | Batch size | |
| AlexNet | 0.70 | SGDM | 0.0001 | 32 |
| VGG-16 | 0.80 | SGDM | 0.0001 | 16 |
| GoogLeNet | 0.75 | ADAM | 0.0001 | 24 |
| ResNet-50 | 0.75 | SGDM | 0.001 | 8 |
Official websites use .gov
A
.gov website belongs to an official
government organization in the United States.
Secure .gov websites use HTTPS
A lock (
) or https:// means you've safely
connected to the .gov website. Share sensitive
information only on official, secure websites.