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/