TorchScript

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

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

能将动态图转为静态图

常和torch.jit合起来用

おすすめ

転載: blog.csdn.net/hxxjxw/article/details/120835884