gpt4 book ai didi

pytorch - torchscript中的torch.jit.trace和torch.jit.script有什么区别?

转载 作者:行者123 更新时间:2023-12-04 13:35:04 44 4
gpt4 key购买 nike

Torchscript 提供了 torch.jit.trace 和 torch.jit.script 将 pytorch 代码从 Eager 模式转换为脚本模型。从文档中,我可以理解 torch.jit.trace 无法处理 Python 中存在的控制流和其他数据结构。因此,torch.jit.script 被开发来克服 torch.jit.trace 中的问题。
但是看起来 torch.jit.script 适用于所有情况,那为什么我们需要 torch.jit.trace 呢?
请帮助我理解这两种方法之间的区别

最佳答案

torch.jit.script适用于您的代码,那么这就是您所需要的。编译器不支持使用动态行为(如多态)的代码 torch.jit.script使用,因此对于此类情况,您需要使用 torch.jit.trace .

关于pytorch - torchscript中的torch.jit.trace和torch.jit.script有什么区别?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62626052/

44 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com