加载预训练模型出错

[DL-Alva]

您好，我在训练过程中加载预训练模型时，报错：RuntimeError: ./weights/yolov5s.pt is a zip archive (did you mean to use torch.jit.load()? 运行环境是cuda:10.0, pytorch:1.5.1, 请问这是怎么回事呢？

[DL-Alva]

我尝试在torch.save 中加入_use_new_zipfile_serialization=False ，但是依旧不起作用

[PeterH0323]

我的运行环境是 cuda 10.2 + pytorch 1.5.1，请问您的整套代码是使用我的吗？我使用YOLOv5的源码是 2.0 的版本

[PeterH0323]

请问您使用的是官方最新的权重文件吗？我估计是因为 YOLOv5 3.0 使用了 pytorch 1.6 的来训练导致官方最新的权重文件和 2.0 版本的不适配了

[DL-Alva]

我用的是您的整套代码，运行环境是cuda10.1+pytorch1.5，我之前看网上的说，我这个报错是pytorch版本导致的，所以我又尝试用了pytorch1.6，依然有这个报错。

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: PeterH0323 <notifications@github.com> 发送时间: 2020年10月16日 23:48 收件人: PeterH0323/Smart_Construction <Smart_Construction@noreply.github.com> 抄送: Kaijun Liu <1536292800@qq.com>, Author <author@noreply.github.com> 主题: 回复：[PeterH0323/Smart_Construction] 加载预训练模型出错 (#7)

我的运行环境是 cuda 10.2 + pytorch 1.5.1，请问您的整套代码是使用我的吗？我使用YOLOv5的源码是 2.0 的版本

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[PeterH0323]

我现在上传权重文件到百度云，之后我会在 README.md 分享出来，我上传成功之后和您说一下

[DL-Alva]

我应该是这周二下载并使用的。我也觉得是因为预训练权重是在pytorch1.6的环境下训练的，和我运行的pytorch版本不匹配。我记得大致的报错内容应该是“加载的模型，也就是pt文件，是压缩文件，没办法处理”

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: PeterH0323 <notifications@github.com> 发送时间: 2020年10月16日 23:55 收件人: PeterH0323/Smart_Construction <Smart_Construction@noreply.github.com> 抄送: Kaijun Liu <1536292800@qq.com>, Author <author@noreply.github.com> 主题: 回复：[PeterH0323/Smart_Construction] 加载预训练模型出错 (#7)

请问您使用的是官方最新的权重文件吗？我估计是因为 YOLOv5 3.0 使用了 pytorch 1.6 的来训练导致了这个问题

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[DL-Alva]

好的，非常感谢您。请问您上传的权重文件是在pytorch1.5环境下训练好的吗？

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: PeterH0323 <notifications@github.com> 发送时间: 2020年10月16日 23:58 收件人: PeterH0323/Smart_Construction <Smart_Construction@noreply.github.com> 抄送: Kaijun Liu <1536292800@qq.com>, Author <author@noreply.github.com> 主题: 回复：[PeterH0323/Smart_Construction] 加载预训练模型出错 (#7)

我现在上传权重文件到百度云，之后我会在 README.md 分享出来，我上传成功之后和您说一下

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[PeterH0323]

我刚刚根据您提供的描述也百度了一下，好像确实是因为权重的问题

[PeterH0323]

是的，上传的权重是经过训练的，具体精度可以看 README.md 😁

[DL-Alva]

如果直接使用pytorch1.6训练好的权重，那您知道这个报错怎么解决吗？之前研究了一天也没找到方法😭

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: PeterH0323 <notifications@github.com> 发送时间: 2020年10月17日 00:02 收件人: PeterH0323/Smart_Construction <Smart_Construction@noreply.github.com> 抄送: Kaijun Liu <1536292800@qq.com>, Author <author@noreply.github.com> 主题: 回复：[PeterH0323/Smart_Construction] 加载预训练模型出错 (#7)

我刚刚根据您提供的描述也百度了一下，好像确实是因为权重的问题

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[DL-Alva]

好的，我明天用您新上传的权重试一下，非常感谢您(⑉• •⑉)

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: PeterH0323 <notifications@github.com> 发送时间: 2020年10月17日 00:03 收件人: PeterH0323/Smart_Construction <Smart_Construction@noreply.github.com> 抄送: Kaijun Liu <1536292800@qq.com>, Author <author@noreply.github.com> 主题: 回复：[PeterH0323/Smart_Construction] 加载预训练模型出错 (#7)

是的，上传的权重是经过训练的，具体精度可以看 README.md 😁

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[PeterH0323]

我之前试过使用 pytorch 1.6 跑我这个模型会出错，当时是有些接口不兼容，之后我退回了 1.5.1 的版本了，如果是用 Pytorch 1.6 的话，可以使用官方最新的 YOLOv5 3.0 版本 + Pytorch 1.6 来训练

[DL-Alva]

好的，我都试一下。谢谢您的回复。

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: PeterH0323 <notifications@github.com> 发送时间: 2020年10月17日 00:10 收件人: PeterH0323/Smart_Construction <Smart_Construction@noreply.github.com> 抄送: Kaijun Liu <1536292800@qq.com>, Author <author@noreply.github.com> 主题: 回复：[PeterH0323/Smart_Construction] 加载预训练模型出错 (#7)

我之前试过使用 pytorch 1.6 跑我这个模型会出错，但是是有些接口不兼容，之后我退回了 1.5.1 的版本了，如果是用 Pytorch 1.6 的话，可以使用官方最新的 YOLOv5 3.0 版本 + Pytorch 1.6 来训练

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[PeterH0323]

权重文件的百度云链接已在 README.md 加入，请查看 😁

[DL-Alva]

好哒😄ཽ

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: PeterH0323 <notifications@github.com> 发送时间: 2020年10月17日 00:21 收件人: PeterH0323/Smart_Construction <Smart_Construction@noreply.github.com> 抄送: Kaijun Liu <1536292800@qq.com>, Author <author@noreply.github.com> 主题: 回复：[PeterH0323/Smart_Construction] 加载预训练模型出错 (#7)

权重文件的百度云链接已在 README.md 加入，请查看 😁

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[PeterH0323]

谢谢您的对我项目的支持，如果解决问题的话我先关闭这个 issue 了，希望能够给您的项目贡献出一份微薄之力，如果帮到您请帮我点个 ✨ ，您的 ✨ 能让技术传播得更远

[DL-Alva]

应该的，也非常感谢您的的耐心解答

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: PeterH0323 <notifications@github.com> 发送时间: 2020年10月17日 00:30 收件人: PeterH0323/Smart_Construction <Smart_Construction@noreply.github.com> 抄送: Kaijun Liu <1536292800@qq.com>, Author <author@noreply.github.com> 主题: 回复：[PeterH0323/Smart_Construction] 加载预训练模型出错 (#7)

谢谢您的对我项目的支持，如果解决问题的话我先关闭这个 issue 了，希望能够给您的项目贡献出一份微薄之力，如果帮到您请帮我点个 ✨ ，您的 ✨ 能让技术传播得更远

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[PeterH0323]

😁

[eeric]

标签做好了，可以使用https://github.com/ultralytics/yolov5训练

[DL-Alva]

好的，非常感谢您☺️

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: eeric <notifications@github.com> 发送时间: 2020年12月11日 22:58 收件人: PeterH0323/Smart_Construction <Smart_Construction@noreply.github.com> 抄送: Kaijun Liu <1536292800@qq.com>, Author <author@noreply.github.com> 主题: 回复：[PeterH0323/Smart_Construction] 加载预训练模型出错 (#7)

标签做好了，可以使用https://github.com/ultralytics/yolov5训练

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.