Releases: bubbliiiing/hrnet-pytorch
Releases · bubbliiiing/hrnet-pytorch
v2.0 -step、cos学习率下降、多GPU训练
仓库特点
- 支持step、cos学习率下降法。
- 支持adam、sgd优化器选择。
- 支持学习率根据batch_size自适应调整。
- 支持不同预测模式的选择,单张图片预测、文件夹预测、视频预测、图片裁剪。
- 更新summary.py文件,用于观看网络结构。
- 增加了多GPU训练。
Weights
v1.0 fix bugs