4

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 呢?

请帮助我了解这两种方法之间的区别

4

1 回答 1

2

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

于 2020-06-29T22:50:47.810 回答