如何在mmdetection训练中断后接着上一次的训练结果训练？

最新推荐文章于 2024-11-14 08:53:59 发布

正在长头发的研究生

最新推荐文章于 2024-11-14 08:53:59 发布

阅读量888

点赞数 2

文章标签： python

本文链接：https://blog.csdn.net/ambition_du/article/details/133294875

版权

如何在mmdetection训练中断后接着上一次的训练结果训练？

我们在训练过程中难免会遇到一些特殊情况，使得训练过程中断，那么我们应该如何在训练中断之后再接着之前的训练结果训练呢？
我们分析train.py中的参数设置
在这里插入图片描述
也就是说，在中断训练后，我们在命令中传入resume-from参数值就可以接着之前的训练，具体命令如下代码所示：

python tools/train.py configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py --work-dir work_dirs/xxx --resume-from work_dirs/xxx/latest.pth

运行之后我们就可以看到模型接着从上次中断的轮数开始训练啦
在这里插入图片描述

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

正在长头发的研究生

关注关注

2
点赞
踩
8

收藏

觉得还不错? 一键收藏
3
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

Swin Transformer实战目标检测：训练自己的数据集

01-04

Transformer发轫于NLP（自然语言处理），并跨界应用到CV（计算机视觉）领域。 Swin Transformer是基于Transformer的计算机视觉骨干网，在图像分类、目标检测、实例分割、语义分割等多项下游CV应用中取得了SOTA的性能。该项工作也获得了ICCV 2021顶会最佳论文奖。本课程将手把手地教大家使用labelImg标注和使用Swin Transformer训练自己的数据集。本课程将介绍Transformer及在CV领域的应用、Swin Transformer的原理。课程以多目标检测（足球和梅西同时检测）为例进行Swin Transformer实战演示。课程在Windows和Ubuntu系统上分别做项目演示。包括：安装软件环境、安装Pytorch、安装Swin-Transformer-Object-Detection、标注自己的数据集、准备自己的数据集（自动划分训练集和验证集）、数据集格式转换（Python脚本完成）、修改配置文件、训练自己的数据集、测试训练出的网络模型、性能统计、日志分析。相关课程：《Transformer原理与代码精讲（PyTorch）》https://edu.csdn.net/course/detail/36697《Transformer原理与代码精讲（TensorFlow）》https://edu.csdn.net/course/detail/36699《ViT（Vision Transformer）原理与代码精讲》https://edu.csdn.net/course/detail/36719《DETR原理与代码精讲》https://edu.csdn.net/course/detail/36768《Swin Transformer实战目标检测：训练自己的数据集》https://edu.csdn.net/course/detail/36585《Swin Transformer实战实例分割：训练自己的数据集》https://edu.csdn.net/course/detail/36586《Swin Transformer原理与代码精讲》 https://download.csdn.net/course/detail/37045

mmdetection断点训练及遇到的问题AttributeError: ‘SSDHead‘ object has no attribute ‘loss_cls‘

TracyGC的博客

05-28

592

由于mmdetection是用终端运行的，如果不小心把终端关掉了，用以下方式继续上次训练.进入mmdetection-main目录里面之后再次重新编译。可以看到train.py里面是有一个resume参数的。

3 条评论您还未登录，请先登录后发表或查看评论

[mmdetection] 如何在训练中断后,接着上次训练?

wzk4869的博客

11-28

962

[mmdetection] 如何在训练中断后,接着上次训练?

【 [mmdetection] 如何在训练中断后,接着上次训练?】

weixin_46600060的博客

03-28

1万+

[mmdetection] 如何在训练中断后,接着上次训练? 最近由于不知名原因,在用 faster rcnn 训练一个大型数据集的时候,在epoch= 20 的时候中断训练了.采用以下方式继续上次训练. 打开 train.py,如图: 也就是说,训练时,最后加一个–resume from 参数,然后后面跟上次训练生成的最后一个权重文件(.pth)就可以了. 因此,在命令行输入以下语句进行训练即可以完全相同的模型配置继续上次的训练啦:: python tools/train.py configs/fas

mmdetection训练中断，怎么接上次训练结果继续训练，或从之前某个epoch的模型直接开始训练？

m0_63947712的博客

04-11

1195

mmdetection 模型训练技巧

Guo_Python的博客

08-21

8288

由于在项目中和比赛中经常用到mmdetection，在此总结一些mmdetection的小技巧，有利于涨点。

MMDetection使用及相关问题

chenfang0529的博客

03-26

4817

搭建mmdetection，并训练自己的数据集，及使用过程中出现的各种问题汇总：服务器缓存，core dumped等

MMdetection训练遇到的各种问题总结

liu_dongd的博客

04-20

3069

label = self.cat2label[name] 报错报错信息如下： KeyError: ‘Traceback (most recent call last):\n File “/home/dreamtech/.conda/envs/AugFPN2/lib/python3.7/site-packages/torch/utils/data/dataloader.py”, line 138, in _worker_loop\n samples = collate_fn([dataset[

MMDetection实战：MMDetection训练与测试

05-09

MMDetection是商汤和港中文大学针对目标检测任务推出的一个开源项目，它基于Pytorch实现了大量的目标检测算法，把数据集构建、模型搭建、训练策略等过程都封装成了一个个模块，通过模块调用的方式，我们能够以很少的...

mmdetection训练SOLOv2修改过的源码

03-19

本文将详细解析如何在mmdetection环境中训练经过修改的SOLOv2模型。首先，我们需要了解mmdetection的基础架构。mmdetection是一个模块化的框架，它允许用户方便地实现、调整和比较各种目标检测方法。其核心组件...

模型训练+人工智能项目+实例分割+mmdetection.rar+完整配置

08-23

【内容概要】: 本资源提供了一个基于mmdetection框架的实例分割项目，压缩包命名为`mmdetection.rar`。项目包含完整的配置文件、预训练模型、数据处理脚本及详细的文档，支持从数据准备到模型训练直至部署的全流程。...

mmdetection最新版说明文档.pdf

09-15

* 可以使用MMDetection在标准数据集上进行推理和训练，也可以在自定义数据集上进行训练和测试。与Detectron2的比较 * MMDetection和Detectron2都是流行的计算机视觉检测工具包，两者之间存在一些差异。 * MM...

mmdetection训练-使用mmdetection训练目标检测算法-支持fasterrcnn+retinanet等算法-项目

10-20

mmdetection训练_使用mmdetection训练目标检测算法_支持fasterrcnn+retinanet等算法_项目源码_优质项目实战

【pytorch】（断点）继续上次训练

胡侃有料的博客

09-20

1793

every blog every motto: You can do more than you think. 0. 前言训练过程中，停止，后续接着训练 1. 正文 1.1 保存信息每个eopch以后需要保存后续接着训练的信息，信息包括，model、optimizer、epoch for epoch in range(start_epoch,end_epoch): for iter ,data in enumerate(dataloader): pass # --------------

深度学习如何恢复训练？中断的训练如何接着之前保存的 ckpt 参数继续训练？Pytorch-Lightning Trainer

weixin_44212848的博客

02-08

3130

加载 ckpt 模型参数继续训练

PyTorch实现断点继续训练

yangling1035的博客

11-02

5764

当epochs值比较大，在训练过程由于一些原因模型停止训练，好的解决方法是（1）重新训练模型；（2）接着上次训练的断点继续训练一、模型的保存与加载 1. 保存整个Module torch.save(net, path) 2. 保存模型参数 state_dict = net.state_dict() torch.save(state_dict , path) 二、模型训练过程中保存 checkpoint = { "net": model.state_dict(), 'op

解决python脚本错误train.py:error: unrecognized arguments