Skip to main content
. 2024 Sep 13;11:1445565. doi: 10.3389/frobt.2024.1445565

TABLE 2.

Optimal hyperparameters of different models for APTOS dataset.

Hyperparameters SwAV EfficientNet-B5 Swin ViT
Optimizer AdamW AdamW AdamW AdamW
Batch size 64 16 32 64
Learning rate 1e3 1e3 1e5 1e5
Weight decay 5e2 5e2 5e2 5e2
Exponential moving avg 0.99 0.99996 0.99 0.99
Epochs 33 35 35 50