网站首页 站内搜索

搜索结果

查询Tags标签: testloader,共有 1条记录
  • ONNXRuntime学习笔记(三)

    接上一篇完成的pytorch模型训练结果,模型结构为ResNet18+fc,参数量约为11M,最终测试集Acc达到94.83%。接下来有分两个部分:导出onnx和使用onnxruntime推理。 一、pytorch导出onnx 直接放函数吧,这部分我是直接放在test.py里面的,直接从dataloader中拿到一个batch的数…

    2022/5/1 6:16:28 人评论 次浏览
扫一扫关注最新编程教程