摘要: 1. 模型导出基础介绍 torch.onnx.export中需要的模型实际上是一个torch.jit.ScriptModule。而要把普通 PyTorch 模型转一个这样的 TorchScript 模型,有跟踪(trace)和记录(script)两种导出计算图的方法。如果给torch.onnx.ex 阅读全文
posted @ 2025-03-04 09:38 ldfm 阅读(10) 评论(0) 推荐(0) 编辑
点击右上角即可分享
微信分享提示