Table 3 Training parameters for each model
From: AI-Dentify: deep learning for proximal caries detection on bitewing x-ray - HUNT4 Oral Health Study
Architecture | Batch size | Learning rate | Max. epochs | Optimiser | Framework |
|---|---|---|---|---|---|
YOLOv5 M | 8 | \(10^{-2}\) | 180 | SGD | PyTorch |
RetinaNet | 4 | \(10^{-5}\) | 200 | Adam | Keras |
EfficientDet D0 | 8 | \(10^{-4}\) | \(10^6\) | AdamW | PyTorch |
EfficientDet D1 | 8 | \(10^{-4}\) | \(10^6\) | AdamW | PyTorch |