【最佳實踐】對多個模型進行訓練時的參數獲取問題

使用itertools中的chain方法將模型中的參數鏈接到一起:

optimizer = optim.Adam(params=chain(backbone_net.parameters(),
                                                linear_rot_net.parameters(),
                                                linear_classify_net.parameters()),
                                   lr=0.09,
                                   weight_decay=1e-4)
發佈了137 篇原創文章 · 獲贊 85 · 訪問量 12萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章