Consider using a constant container instead (e.g. for `list`, use a `tuple` instead. for `dict`, us

编程知识 更新时间:2023-05-02 05:34:18

使用torch.jit.trace导出模型时出现:

RuntimeError: Encountering a dict at the output of the tracer might cause the trace to be incorrect, this is only valid if the container structure does not change based on the module's inputs. Consider using a constant container instead (e.g. for `list`, use a `tuple` instead. for `dict`, use a `NamedTuple` instead). If you absolutely need this and know the side effects, pass strict=False to trace() to allow this behavior

按照提示的意思进行修改:

before:

script_module = torch.jit.trace(model,x)

修改后:

script_module = torch.jit.trace(model,x,strict=False)

究其原因是因为输出结果中出现了list或者dic

 

更多推荐

Consider using a constant container instead (e.g. for `list`, use a `tuple` inst

本文发布于:2023-04-26 07:16:00,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/48d5c381cef61f43bac1732c9836f002.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:container   constant   list   dict   tuple

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!

  • 104672文章数
  • 26217阅读数
  • 0评论数