登录 注册
当前位置:主页 > 资源下载 > 9 > Rust绑定 for TVM运行时: (合并)

Rust绑定 for TVM运行时: (合并)

  • 更新:2024-09-14 16:13:32
  • 大小:47KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

已淘汰 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