torch.jit.frontend.UnsupportedNodeError: function definitions aren‘t supported

最新推荐文章于 2024-06-06 11:38:52 发布

AI算法网奇

最新推荐文章于 2024-06-06 11:38:52 发布

阅读量5.3k

点赞数 1

分类专栏： c++入门宝典 pytorch知识宝典文章标签： pytorch 深度学习人工智能

本文链接：https://blog.csdn.net/jacke121/article/details/93752707

版权

pytorch知识宝典同时被 2 个专栏收录

512 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

c++入门宝典

359 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文讨论了在使用PyTorch的Just-In-Time (JIT) 编译时遇到的'UnsupportedNodeError: function definitions aren't supported'错误。错误原因可能与代码对齐有关，特别是def __init__(self)函数的缩进。解决方法包括调整代码对齐或者临时注释掉@torch.jit.script装饰器。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch.jit.frontend.UnsupportedNodeError: function definitions aren't supported

 class MyModule(torch.jit.ScriptModule):
        __constants__ = ['mods']

        def __init__(self):
            super(MyModule, self).__init__()
            self.mods = torch.nn.ModuleList([SubModule() for i in range(10)])

  
   错误写法：
    @torch.jit.script_method
    def forward(self, x, targets=None):
        def _branch(_embedding, _in):
            for i, e in enumerate(_embedding):
                _in = e(_in)
                if i == 4:
                    out_branch = _in
            return _in, out_branch
        #  backbone
        x2, x1, x0 = self.backbone(x.cuda())

了解本专栏