本發(fā)明公開了目標檢測模型學習率的設置方法和裝置,涉及計算機技術領域。該方法的一具體實施方式包括設置訓練集總圖片數(shù)量和每輪迭代訓練的圖片數(shù)量,以計算步長;初始化學習率,以在訓練集上進行目標檢測模型的訓練,且每迭代一個步長將學習率增大預設數(shù)值,同時計算一次驗證集的平均精度均值,進而獲得最小學習率和最大學習率;再以預設數(shù)量個步長為一個周期循環(huán)訓練目標檢測模型,直到完成預設的最大迭代次數(shù);其中,基于最小學習率和最大學習率,根據(jù)步長內(nèi)當前迭代的次數(shù),計算每次迭代的學習率。從而,本發(fā)明的實施方式能夠解決現(xiàn)有技術訓練目標檢測模型的學習率設置效率低的問題。
聲明:
“目標檢測模型學習率的設置方法和裝置” 該技術專利(論文)所有權利歸屬于技術(論文)所有人。僅供學習研究,如用于商業(yè)用途,請聯(lián)系該技術所有人。
我是此專利(論文)的發(fā)明人(作者)