Releases: bubbliiiing/classification-keras
Releases · bubbliiiing/classification-keras
v2.0 -step、cos学习率下降、多GPU训练、新增Auto Aug等
重要更新
- 支持step、cos学习率下降法。
- 支持adam、sgd优化器选择。
- 支持学习率根据batch_size自适应调整。
- 更新summary.py文件,用于观看网络结构。
- 新增Auto Aug。
- 增加了多GPU训练。
v1.0 -不同尺寸模型、多个可调整参数。
仓库特点
- 支持不同模型训练。
- 大量的注释,多个可调整参数。
- 支持top1-top5的准确度评价。
- summary.py文件用于观看网络结构。