Skip to content

Releases: bubbliiiing/classification-pytorch

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

22 Apr 12:11
Compare
Choose a tag to compare

重要更新

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

v1.0 -不同尺寸模型、多个可调整参数。

22 Mar 13:04
Compare
Choose a tag to compare

仓库特点

  • 支持不同模型训练。
  • 大量的注释,多个可调整参数。
  • 支持top1-top5的准确度评价。
  • summary.py文件用于观看网络结构。