什么是TorchScript

$$什么是TorchScript$$

---

Pytorch中文文档

---

TorchScript是PyTorch模型（nn.Module的子类）的中间表示，可以在高性能环境（例如C ++，注意不止是c++）中运行。注意的是TorchScript是可以通过python语言使用和导出的。

Pytorch模型转TorchScrip进行保存，保存后的pt模型可以在C++等高性能环境中运行

net.eval()
trace = torch.jit.trace(net, torch.randn(1, 3, 512, 512))
torch.jit.save(trace,'UNet_model.pt')

将模型加载到C ++中以实现不依赖Python的执行

额外补充

一 在C++中加载TorchScript模型

二 PyTorch C ++ API文档

三 PyTorch Python API文档