2022年5月1日
ONNXRuntime学习笔记(三)
摘要： 接上一篇完成的pytorch模型训练结果，模型结构为ResNet18+fc，参数量约为11M，最终测试集Acc达到94.83%。接下来有分两个部分：导出onnx和使用onnxruntime推理。 ##一、pytorch导出onnx 直接放函数吧，这部分我是直接放在test.py里面的，直接从data 阅读全文
posted @ 2022-05-01 00:06 Lee-zq 阅读(3351) 评论(0) 推荐(0) 编辑