YOLO v4 darknet在Google Colab训练中断 如何获取已训练权重
Google Colab训练中断后恢复已训练权重的解决方案
- 若你训练前未将Google Drive挂载到运行时,也没有在训练配置中指定将中间权重保存到云盘路径,无法找回已训练的权重。Colab免费/基础版运行时的存储是临时属性,运行时断开、被平台回收、切换运行时后,所有运行时内生成的文件(包括克隆的darknet仓库、中间训练权重、临时数据集文件)都会被永久清空,没有恢复途径。
- 若你此前已经完成了云盘挂载,且训练命令中设置了权重保存路径为云盘目录,可按以下步骤找回:
- 新建笔记本重新挂载云盘,执行代码:
from google.colab import drive drive.mount('/content/drive')- 进入你之前指定的权重保存目录,查找后缀为
.weights的文件即可,darknet默认每100轮epoch自动保存一次中间权重,如果你训练时添加了-save_interval N参数,会按你设定的每N轮保存一次。
后续训练避坑建议
- 训练启动前先挂载谷歌云盘,将darknet仓库、数据集、权重保存路径都设置在云盘目录下,避免运行时回收导致文件丢失
- 训练命令中明确指定权重保存路径和保存间隔,避免长时间训练成果丢失
内容的提问来源于stack exchange,提问作者Nakkhatra
相关产品推荐
相关产品推荐

