Skip to content

v2.0 -step、cos学习率下降、多GPU训练、新增Auto Aug等

Latest
Compare
Choose a tag to compare
@bubbliiiing bubbliiiing released this 22 Apr 12:11
· 6 commits to main since this release

重要更新

  • 支持step、cos学习率下降法。
  • 支持adam、sgd优化器选择。
  • 支持学习率根据batch_size自适应调整。
  • 更新summary.py文件,用于观看网络结构。
  • 新增Auto Aug。
  • 增加了多GPU训练。