
使用PyTorch构建卷积神经网络进行手写识别,并将训练好的模型部署至TensorRT以实现加速
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目利用PyTorch框架建立卷积神经网络模型,用于精准识别手写数字。通过优化与TensorRT集成,显著提升了模型的运行效率和速度。
使用PyTorch构建卷积神经网络以完成手写识别任务,并将训练好的模型通过多种方式部署到TensorRT中进行加速。(1)通过加载权重;(2)通过加载ONNX格式的模型文件;(3)直接加载Engine文件。
全部评论 (0)
还没有任何评论哟~


