|
1 |
| -# TVM runtime frontend support (pre-release) |
| 1 | +# TVM Runtime Frontend Support |
2 | 2 |
|
3 |
| -This crate provides idiomatic Rust API for [TVM](https://github.com/dmlc/tvm) runtime frontend as part of [RFC 1601](https://github.com/dmlc/tvm/issues/1601). Currently this requires **Nightly Rust**. |
| 3 | +This crate provides idiomatic Rust API for [TVM](https://github.com/dmlc/tvm) runtime frontend as part of [ongoing RFC 1601](https://github.com/dmlc/tvm/issues/1601). Currently this requires **Nightly Rust**. |
4 | 4 |
|
5 |
| -Checkout the [docs](https://ehsanmok.github.io/tvm_frontend/tvm_frontend/index.html) |
| 5 | +Checkout the [docs](https://ehsanmok.github.io/tvm_frontend/tvm_frontend/index.html). |
6 | 6 |
|
7 |
| -## What does this crate offer? |
| 7 | +## What Does This Crate Offer? |
8 | 8 |
|
9 | 9 | Here is a major workflow
|
10 | 10 |
|
11 |
| -1. Train your Deep Learning model using any major deep learning framework [PyTorch](https://pytorch.org/), [Apache MXNet](https://mxnet.incubator.apache.org/) and [TensorFlow](https://www.tensorflow.org/) |
12 |
| -2. Use TVM to build optimized model artifacts on a given supported TVM context such as CPU, GPU, OpenCL, Vulkan, VPI, ROCM, etc. |
13 |
| -3. Deploy your models using Rust :heart: |
| 11 | +1. Train your **Deep Learning** model using any major framework [PyTorch](https://pytorch.org/), [Apache MXNet](https://mxnet.incubator.apache.org/) and [TensorFlow](https://www.tensorflow.org/) |
| 12 | +2. Use **TVM** to build optimized model artifacts for a given supported TVM context such as CPU, GPU, OpenCL, Vulkan, VPI, ROCM, etc. |
| 13 | +3. Deploy your models using **Rust** :heart: |
14 | 14 |
|
15 |
| -### Example: Resnet18 pretrained on Imagenet |
| 15 | +### Example: Deploy Image Classification from Pretrained Resnet18 on ImageNet1k |
16 | 16 |
|
17 | 17 | Please checkout [examples/resnet](https://github.com/ehsanmok/tvm-rust/tree/master/examples/resnet) for the complete end-to-end example.
|
18 | 18 |
|
19 |
| -Here's python snippet for download and building Resnet18 via MXNet and TVM |
| 19 | +Here's python snippet for download and building a pretrained Resnet18 via MXNet and TVM |
20 | 20 |
|
21 | 21 | ```python
|
22 | 22 | block = get_model('resnet18_v1', pretrained=True)
|
@@ -100,7 +100,7 @@ Please follow the TVM [installation](https://docs.tvm.ai/install/index.html), `e
|
100 | 100 |
|
101 | 101 | *Note:* To run the end-to-end examples and tests, `tvm`, `nnvm` and `topi` need to be added to your `PYTHONPATH`.
|
102 | 102 |
|
103 |
| -## Other supported functionalities |
| 103 | +## Supported TVM Functionalities |
104 | 104 |
|
105 | 105 | ### Use TVM to Generate Shared Library
|
106 | 106 |
|
|
0 commit comments