使用pytorch torch.jit.trace转换模型时出错“xxx has no attribute xxx“

最新推荐文章于 2024-05-15 16:36:35 发布

CV蒸汽机

最新推荐文章于 2024-05-15 16:36:35 发布

阅读量1.6k

点赞数

本文链接：https://blog.csdn.net/dabian1793/article/details/114310868

版权

在使用torch.jit.trace方法转换模型时，对于方法中的if和循环控制流是没办法处理的，这样就会出现只要产生该模块时默认都会有这个参数，但是在有些利用if来控制模型结构的地方是没有这个属性的，因此报错。
解决办法：利用torch.jit.script方法转换

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

CV蒸汽机

关注关注

0
点赞
踩
0

收藏

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

分享到 QQ

分享到新浪微博

扫一扫

module 'torch.jit' has no attribute 'unused'

jacke121的专栏

03-10

1万+

module 'torch.jit' has no attribute 'unused' 解决方法：安装新版torch： pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

使用torch.jit.trace另存模型为TorchScript格式出现的问题

最新发布

weixin_46986751的博客

06-04

631

②检查模型导入时的设置，有些模型训练和测试调用的forward函数不同，导入模型时只需将相关参数设置成训练即可，以ssd.pytorch为例。像上述ssd返回的就是一个list，可以改成return output[0]，返回第一个值即可解决。，可以将后处理过程单独定义一个函数，模型推理完再调用该函数进行后处理；①将后处理过程（Detect、NMS等，具体看报错的是哪个函数）从。解决方式：检查模型forward函数的返回值是否为Tensor。字面解释：无法导出“Detect”函数，需要将其移除。

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

pytorch1.4 导出模型问题记录

weixin_45650404的博客

03-06

1733

在pytorch1.4，使用torch.jit.trace方法导出YOLO3模型时，执行traced_script_module.save("yolov3.pt")保存后，在C++环境下进行载入测试，载入的模型参数都正确，不知什么原因输出却不对。最后将尝试将保存模型名称后缀改为.pth后traced_script_module.save("yolov3.pth"),输出正常。 ...

关于torch.jit.trace在yolov8中出现的问题

abcdzgh的博客

06-29

545

1）为什么yolov8不能直接torch.jit.trace，需要经过图像检测后才能，且检测后self.net发生变化；而2中的第一版yolov5可以直接torch.jit.trace2）3中的第二版yolov5初始不能直接torch.jit.trace，用了两种方式后，都能了，但两种方式却不是相同的，那他们哪一点共同点，导致都能转模型了呢。

pytorch torch.jit.trace()使用模型转换要注意的问题

xingtianyao的博客

09-28

1万+

https://www.jianshu.com/p/8206b198533d 如果代码中有if条件控制，尽量避免使用torch.jit.trace来转换代码，因为它不能处理变化条件，如果非要用trace的话，可以把if条件控制改成别的形式，比如： def f(x): if x > 0: return False else: return True 可以改成: def f(x): return x <= 0 jit不能转换第三方Py...

导出torchscript过程中报错‘method_descriptor‘ object has no attribute ‘__globals__‘或者“undefined value F”

weixin_44415029的博客

02-23

1207

出现这两个问题的原因是python版本问题。python 3.9目前使用libtorch的script方法或者trace方法导出torchscript会导致这个难以解决的bug。将python版本更换为python3.6即可。

Pytorch JIT 使用经验总结

The space of Shining

01-26

7930

注意所有的torch.operator，需要传dim的。如 torch.cat(tensors, dim=0, out=None) → Tensor 传参数必要要注意，dim绝对不能=-1，因为使用Trace的时候，CPU虽然可能一切正常，但是使用GPU进行Trace的时候会报错。一般会报如下错误： torch.jit.TracingCheckError: Tracing failed san...

pytorch JIT浅解析

热门推荐

The space of Shining

01-16

4万+

概要 Torch Script中的核心数据结构是ScriptModule。它是Torch的nn.Module的类似物，代表整个模型作为子模块树。与普通模块一样，ScriptModule中的每个单独模块都可以包含子模块，参数和方法。在nn.Modules中，方法是作为Python函数实现的，但在ScriptModules方法中通常实现为Torch Script函数，这是一个静态类型的Pyt...

解决错误：AttributeError：model ’torch.jit. has no attribute ‘unused‘

ZwaterZ的博客

06-13

2450

导入torchvision出现：AttributeError: module ‘torch.jit’ has no attribute 'unused’错误报错截图：网上查找资料得，错误原因说由于torch和torchvision的版本导致的该错误。加载对应的pytorch和torchvision 本人使用torch== 1.14.0，需加载torchvision== 0.5.0...

AttributeError: module ‘torch.jit‘ has no attribute ‘_script_if_tracing‘

Hawley的博客

11-06

1万+

AttributeError: module ‘torch.jit‘ has no attribute ‘_script_if_tracing‘ #今天在GitHub上找了一个CNN的项目，准备跑一下，然后报这么一个错下面记录一下。首先我百度到了这篇博客：https://blog.csdn.net/qq_38343151/article/details/108570528 第一条命令是删除当前环境中的torchvision 试了一下，成功删除了但是等到输入第二条命令的时候，报错了，然后我接着试了这么一

pytorch保存模型参数信息与模型结构信息

Grandret的博客

11-19

3421

阅读官网文档发现pytorch除了使用torch.save与torch.load保存加载数据/模型外还可以使用torch.jit.save(),torch.jit.load()。后者的好处在于其他py脚本在使用模型时不需要将原模型所在py文档也引入可以跨平台使用官网代码 scripted_module = torch.jit.script(MyModule()) torch.jit.save(scripted_module, 'mymodule.pt') torch.jit.load('mymo

初学python遇到PyTorch AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘的问题

weixin_45806487的博客

12-20

9549

记录原因：初次使用AI，应用一些有趣简单的代码使用场景：使用网上下载的代码初次使用网上的代码，重要的是要引入第三方库以及路径问题描述：安装库使用requirements.txt安装库,在Anaconda输入。 python -m pip install -r G:\requirements.txt 注：requirement.txt文件在G盘，-r是读取文件，-m是在以脚本模式启动，详情： https://www.cnblogs.com/xueweihan/p/5118222.html

Pytorch模型转成onnx并可视化

小威的博客

01-03

2269

本文介绍 pth 模型转为 onnx 模型、使用 onnx 模型进行可视化以及过程中可能出现的问题。

torch.jit.trace与checkpoint.checkpoint冲突：Could not export Python function call ‘CheckpointFunction‘.

weixin_42421591的博客

03-06

474

torch.jit.trace与checkpoint.checkpoint冲突：Could not export Python function call 'CheckpointFunction'.

torch.jit.trace详解

qq_44815135的博客

05-15

1284

torch.jit.trace() 是 PyTorch 中用于模型跟踪（tracing）的函数。跟踪是将模型转换为 TorchScript 表示的过程，它将模型的操作序列记录下来，并生成一个可以在 TorchScript 环境下运行的优化版本。func：一个包含模型前向传播逻辑的函数，可以是模型的 forward() 方法或任何能够在输入上执行前向传播的函数。example_inputs：模型的示例输入，用于跟踪过程。它可以是单个张量、元组或列表，用于匹配模型接受的输入。

错误'dict' object has no attribute 'forward'的解决办法

pluto21的博客

08-27

1万+

出现问题主要是因为实例化画笔时写的为： turtle.pen 应改为： turtle.Pen 其中P需大写

AttributeError: ‘RecursiveScriptModule‘ object has no attribute ‘copy‘

我的小天地 ^_^

06-08

5221

在使用pytorch的pt文件推理时候，出现了报错，可以通过本文方法解决

torch.jit.trace 消除TracerWarning

驽马十驾，功在不舍；锲而不舍，金石可镂。

06-19

1万+

torch.jit.trace TracerWarning技巧1.尽量不要用tensor.shape/torch.size()1.1 生成新的tensor1.2 if/while语句中2.比较两个单元素的tensor3.使用参数strict=False4.其他参考文献在使用torch.jit.trace时，经常会碰到如下warning： TracerWarning: Converting a tensor to a Python boolean might cause the trace to be i

AttributeError: module 'torch.jit' has no attribute 'compile'

07-28

非常抱歉，我之前给出的回答有误。在 PyTorch 中，确实没有名为 `torch.compile` 的函数。如果你想要将模型编译为 TorchScript 格式，正确的方法是使用 `torch.jit.script` 函数。这个函数将模型转换为 TorchScript，并返回一个可调用的 TorchScript 模块。下面是一个示例代码： ```python import torch import torchvision.models as models # 加载一个预训练的模型 model = models.resnet18(pretrained=True) # 将模型转换为 TorchScript scripted_model = torch.jit.script(model) # 使用 TorchScript 模块进行推理 input_tensor = torch.randn(1, 3, 224, 224) output = scripted_model(input_tensor) ``` 请注意，`torch.jit.script` 函数可以将大多数的 PyTorch 模型转换为 TorchScript，但也有一些限制。你可以查阅 PyTorch 官方文档中关于 TorchScript 的部分，了解更多关于转换模型和使用 TorchScript 的详细信息。如果你有其他问题或需要进一步帮助，请随时提问。