Table 4.
Adjust training parameters options for AlexNet, GoogLeNet and ResNet-18 models.
Options | AlexNet | GoogleNet | ResNet-18 |
---|---|---|---|
training Options | adam | adam | adam |
Mini Batch Size | 130 | 20 | 15 |
Max Epochs | 10 | 5 | 8 |
Initial Learn Rate | 0.0001 | 0.0003 | 0.0001 |
Validation Frequency | 50 | 3 | 5 |
Execution Environment | 4 GB GPU | 4 GB GPU | 4 GB GPU |