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