Skip to content

v3.1 -修改loss组成、step、cos学习率下降、多GPU训练等

Compare
Choose a tag to compare
@bubbliiiing bubbliiiing released this 23 Feb 12:32
· 11 commits to main since this release

重要更新

  • 修改了loss组成,使得分类、目标、回归loss的比例合适。
  • 支持step、cos学习率下降法。
  • 支持adam、sgd优化器选择。
  • 支持学习率根据batch_size自适应调整。
  • 支持不同预测模式的选择,单张图片预测、文件夹预测、视频预测、图片裁剪、heatmap、各个种类目标数量计算。
  • 更新summary.py文件,用于观看网络结构。
  • 增加了多GPU训练。