![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
模型处理
努力下班
这个作者很懒,什么都没留下…
展开
-
从.pth文件统计神经网络模型参数量
对于网络模型未知且只有.pth参数的网络模型,下载.pth文件后,可采用以下方法统计网络模型参数。原创 2024-06-20 20:13:11 · 155 阅读 · 0 评论 -
导出onnx模型(2):torch.onnx.export 单输入
将模型的输入直接传输到torch.onnx.export()函数中即可。原创 2024-05-16 09:22:25 · 118 阅读 · 0 评论 -
导出onnx模型(1):torch.onnx.export 多输入
将模型的多个输入构造为dict,输入到torch.onnx.export()函数中,实现多输入模型导出onnx。原创 2024-05-06 10:40:10 · 460 阅读 · 0 评论