- 博客(1)
- 收藏
- 关注
RSS订阅原创 使用TVM编译自己的简单模型
TVM-LeNet本文完成了自己搭建、训练和打包LeNet模型，并用TVM对其进行简单的优化，比较两者的异同，发现TVM的计算图更为简捷，准确率二者几乎相同，并且TVM的推理速度是普通模型的2倍左右。版本：TVM 0.7.dev1 ; pytorch 1.6.0 ; GTX1650 , cuda10.2文章目录TVM-LeNet@[toc]1.LeNet模型搭建：2.模型训练：3.模型推理：4.使用TVM进行推理：5.结果比较：1.LeNet模型搭建：LeNet-5是1998年Yann Lecun
2021-03-12 17:13:10
971
5
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人