Skip to main content
. 2021 Apr 21;134:104401. doi: 10.1016/j.compbiomed.2021.104401

Table 2.

The range of hyper-parameters used for different pre-trained models.

Deep Model Batch Size Learning Rate
AlexNet 48, 32 5e-10, 1e-8, 8e-7,
6e-7, 1e-7, 1e-6, 1e-5
SqueezeNet 48, 32, 24 1e-8, 8e-7, 1e-7,
1e-6, 9e-5
DenseNet201 48, 32 1e-7, 1e-6,1e-5
MobileNetV2 48, 32, 24 1e-8, 6e-7, 4e-7, 1e-7,
9e-6, 5e-6, 3e-6, 1e-6, 8e-5
InceptionV3 36, 32, 24 1e-8, 1e-6, 8e-5, 2e-5,
1e-5
Our ensemble model (InceptionV3 + MobileNetV2) 48 9e-8, 9e-7, 7e-7, 4e-7, 1e-7,
7e-6, 1e-6