pytorch 通用训练代码讲解(very good)

文章目录

    • 1. 模型训练参数设置
    • 2. 保证模型可复现性
    • 3. 设置device
    • 4 初始化模型权重及加载预训练权重
      • 4.1 初始化模型权重
      • 4.2 加载预训练权重
      • 4.3 在线下载预训练权重
    • 5 k-means 聚类anchors
    • 6 多卡同步bn及并行运行
    • 7 权重指数平滑ModelEMA
      • 7.1 EMA的原理及作用
      • 7.2 EMA的实现
      • 7.2 EMA的使用
    • 8 optimizer设置

你可能感兴趣的:(object,detection,pytorch,yolo,模型训练)