Pytorch torchscript中torch.jit.trace和torch.jit.script之间有什么区别?
Torchscript提供torch.jit.trace和torch.jit.script,以将pytorch代码从急切模式转换为脚本模型。从文档中,我可以理解torch.jit.trace无法处理python中的控制流和其他数据结构。因此,开发torch.jit.script是为了克服torch.jit.trace中的问题 但看起来torch.jit.script适用于所有情况,那么为什么我们需要torch.jit.trace?Pytorch torchscript中torch.jit.trace和torch.jit.script之间有什么区别?,pytorch,jit,torchscript,Pytorch,Jit,Torchscript,Torchscript提供torch.jit.trace和torch.jit.script,以将pytorch代码从急切模式转换为脚本模型。从文档中,我可以理解torch.jit.trace无法处理python中的控制流和其他数据结构。因此,开发torch.jit.script是为了克服torch.jit.trace中的问题 但看起来torch.jit.script适用于所有情况,那么为什么我们需要torch.jit.trace? 请帮助我理解这两种方法之间的区别如果torch.jit.scri
请帮助我理解这两种方法之间的区别如果
torch.jit.script
适用于您的代码,那么这就是您所需要的。编译器torch.jit.script
uses不支持使用动态行为(如多态性)的代码,因此对于这种情况,您需要使用torch.jit.trace