TorchScript

TorchScript,第1张

TorchScript

Torchscript是Pytorch模型(继承自nn.Module)的中间表示,可以在像C++这种高性能的环境中运行

Torchscript是一种从PyTorch代码创建可序列化和可优化模型的方法。任何Torchscript程序都可以从Python进程中保存,并加载到没有Python依赖的进程中。

能将动态图转为静态图

常和torch.jit合起来用

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/4655436.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-11-06
下一篇 2022-11-06

发表评论

登录后才能评论

评论列表(0条)

保存