2020年6月7日
基于PyTorch与TensorRT的cifar10推理加速引擎(C++)
摘要： 一、写在开头 1、基于PyTorch训练出cifar10模型 2、以ONNX(Open Neural Network Exchange)格式导出模型cifar10.onnx 3、下载cifar10二进制版本数据集 4、创建TensorRT(vs c++)项目，解析模型，进行推理 二、基于PyTorc 阅读全文
posted @ 2020-06-07 19:55 卡冈图雅blackhole 阅读(1249) 评论(0) 推荐(0) 编辑