网络模型 | 学习率 | 批量大小 | 优化器 | 学习率调整机制 | 权值 衰减 | 丢弃率 | 训练最大轮次 | 输入图片大小 | 数据集 |
ResNet-50 | 0.0001 | 64 | 0.9/0.999 | 0.96 | 0.00001 | 0.3 | 100 | 64 × 64/224 × 224 | IP102, D0 |
FPNs | 0.0001 | 32 | 0.9/0.999 | 0.96 | 0.00001 | 0 | 100 | 64 × 64/224 × 224 | IP102, D0 |
WraNet (IP102) | 0.01 | 128 | 0.9/0.999 | 0.96 | 0.00001 | 0.3 | 150 | 64 × 64 | IP102 |
WraNet (D0) | 0.001 | 128 | 0.9/0.999 | 0.96 | 0.0001 | 0.3 | 100 | 64 × 64 | D0 |
SqueezeNet1_1 | 0.0001 | 64 | 0.9/0.999 | 0.96 | 0.0002 | 0.5 | 100 | 64 × 64 | IP102、D0 |
MobileNetV3 | 0.0001 | 64 | 0.9/0.999 | 0.96 | 0.00004 | 0.2 | 100 | 64 × 64 | IP102、D0 |
shufflenet_v2 | 0.0001 | 64 | 0.9/0.999 | 0.96 | 0.0001 | 0.1 | 100 | 64 × 64 | IP102、D0 |
EfficientNet-B0 | 0.0001 | 64 | 0.9/0.999 | 0.96 | 0.0001 | 0.3 | 100 | 64 × 64 | IP102、D0 |