-
Rust绑定 for TVM运行时: (合并)
资源介绍
已淘汰
RFC已关闭,并且已合并到。
TVM运行时前端支持
此板条箱为运行时前端提供了惯用的Rust API,作为 。 目前,这需要Nightly Rust 。
签出。
此板条箱提供什么?
这是主要的工作流程
使用 , 或等任何主要框架训练您的深度学习模型
使用TVM在受支持的上下文(例如CPU,GPU,OpenCL,Vulkan,VPI,ROCM等)上构建优化的模型工件。
使用Rust部署模型 :red_heart_selector:
示例:从ImageNet1k上的预训练Resnet18部署图像分类
请查看以获取完整的端到端示例。
这是一个Python片段,用于通过MXNet和TVM下载和构建经过预训练的Resnet18
block = get_model ( 'resnet18_v1' , pretrained = True )
sym , params = nnvm . frontend . from_mxnet ( block )
# add the softmax layer for prediction
net = nnvm . sym . softm
- 上一篇: 取色器 获得颜色代码
- 下一篇: 第八课:机器学习模型用于评分卡模型-GBDT