pytorch 模型不一樣層設置不一樣的學習率

  在目標檢測的模型訓練中， 咱們一般都會有一個特徵提取網絡backbone， 例如YOLO使用的darknet SSD使用的VGG-16。python 爲了達到比較好的訓練效果， 每每會加載預訓練的backbone模型參數， 而後在此基礎上訓練檢測網絡， 並對backbone進行微調， 這時候就須要爲backbone設置一個較小的lr。git class net(torch.nn.Module)

>>阅读原文<<