pytorch 模型不一样层设置不一样的学习率

  在目标检测的模型训练中, 咱们一般都会有一个特征提取网络backbone, 例如YOLO使用的darknet SSD使用的VGG-16。python 为了达到比较好的训练效果, 每每会加载预训练的backbone模型参数, 而后在此基础上训练检测网络, 并对backbone进行微调, 这时候就须要为backbone设置一个较小的lr。git class net(torch.nn.Module)
相关文章
相关标签/搜索