網站首頁 編程語言 正文
YOLOV5-斷點訓練/繼續訓練
情況描述:
1.訓練過程中中斷了,繼續訓練
2.訓練完了,但是未收斂,在這個基礎上,還想用這個權重、學習率等參數繼續訓練多一些批次
1. 訓練過程中中斷了,繼續訓練
斷電、或者什么原因中斷了,比如設定epoch為200,但是在90這里中斷了,想從斷點這里繼續 訓練到 200個epoch
直接在trainds.py設置參數–resume的defualt=True即可(在代碼 517行附近)
1.parser.add_argument('--resume', nargs='?', const=True, default=True, help='resume most recent training')
2. 訓練完原有epoch,發現沒有收斂,想繼續訓練
訓練完原有epoch后,但還繼續訓練,比如設置epoch為30,已經訓練完了,但是沒有收斂,想使用訓練了30 epoch的權重繼續訓練40個epoch, 總共就是70個epoch
**注意:**不是直接在train.py設置weight參數為訓練好的權重(last.pt),然后設置epoch為70,這樣不行,因為學習率等超參數啟動時還是按照默認的。(自己把一些超參數設置好也可以,比較麻煩)
在train.py代碼里面設置就可以使用原有30 epoch的權重后繼續訓練到40epoch
- step1:
先將–resume的defualt改為True,再將–epochs設為想要繼續訓練到的epochs,即defualt=70(在trainds.py代碼的513行附近)
parser.add_argument('--epochs', type=int, default=70)
- step2:
在trainds.py文件中,按下Ctrl+F搜索Epochs定位到大概197行的位置,做以下兩處修改:
添加一行代碼:ckpt[‘epoch’] = 30
修改代碼:start_epoch = ckpt[‘epoch’](后面不加1)
注意:這兩處修改是為了斷點訓練,如果不再需要斷點訓練,請將這兩處修改都還原。
- step3:
在trainds.py文件中,按下Ctrl+F搜索replace定位到大概565行的位置,做以下兩處修改:
如圖添加一行代碼:init_epochs = opt.epochs(記錄最開始由epochs參數設定的數值70,因為replace這行代碼,將epochs替換成了原始的epochs數值30)
修改代碼:如圖進行添加,reinstate恢復成設定的epochs
做完以上3個修改,訓練就會從第30個epoch的基礎上進行繼續訓練到70個epoch,和直接設置epoch為70一樣
原文鏈接:https://blog.csdn.net/weixin_39589455/article/details/128461081
- 上一篇:沒有了
- 下一篇:沒有了
相關推薦
- 2022-06-18 android實現在圖標上顯示數字_Android
- 2022-12-04 Jetpack之CameraX的使用_Android
- 2023-07-13 websocket的使用及nginx通信的ws代理配置
- 2022-09-22 Laravel解決高并發問題的思路
- 2022-11-13 詳解Redis?Stream做消息隊列_Redis
- 2022-10-09 React高階組件的使用淺析_React
- 2022-05-10 怎樣理解單項數據流
- 2022-07-11 atom插件之·atom-minify配置支持壓縮es6
- 欄目分類
-
- 最近更新
-
- window11 系統安裝 yarn
- 超詳細win安裝深度學習環境2025年最新版(
- Linux 中運行的top命令 怎么退出?
- MySQL 中decimal 的用法? 存儲小
- get 、set 、toString 方法的使
- @Resource和 @Autowired注解
- Java基礎操作-- 運算符,流程控制 Flo
- 1. Int 和Integer 的區別,Jav
- spring @retryable不生效的一種
- Spring Security之認證信息的處理
- Spring Security之認證過濾器
- Spring Security概述快速入門
- Spring Security之配置體系
- 【SpringBoot】SpringCache
- Spring Security之基于方法配置權
- redisson分布式鎖中waittime的設
- maven:解決release錯誤:Artif
- restTemplate使用總結
- Spring Security之安全異常處理
- MybatisPlus優雅實現加密?
- Spring ioc容器與Bean的生命周期。
- 【探索SpringCloud】服務發現-Nac
- Spring Security之基于HttpR
- Redis 底層數據結構-簡單動態字符串(SD
- arthas操作spring被代理目標對象命令
- Spring中的單例模式應用詳解
- 聊聊消息隊列,發送消息的4種方式
- bootspring第三方資源配置管理
- GIT同步修改后的遠程分支