parent
454d7fdc46
commit
466ed64739
|
@ -122,6 +122,5 @@ def test_basesegmentor_forward():
|
|||
We provide some utilities to test rewritten functions. At first, you can construct a model and call `get_model_outputs` to get outputs of the original model. Then you can wrap the rewritten function with `WrapModel`, which serves as a partial function, and get the results with `get_rewrite_outputs`. `get_rewrite_outputs` returns two values that indicate the content of outputs and whether the outputs come from the backend. Because we cannot assume that everyone has installed the backend, we should check if the results are generated by a Python or backend engine. The unit test must cover both conditions. Finally, we should compare the original and rewritten outputs, which may be done simply by calling `torch.allclose`.
|
||||
|
||||
## Note
|
||||
TODO: Add a link to api document after we have a website.
|
||||
|
||||
To learn the complete usage of the test utilities, please refer to our [api document](unit_test_api_link).
|
||||
To learn the complete usage of the test utilities, please refer to our apis document.
|
||||
|
|
|
@ -5,7 +5,7 @@
|
|||
- 下载代码仓库 MMDeploy
|
||||
|
||||
```bash
|
||||
git clone -b master git@github.com:grimoire/deploy_prototype.git MMDeploy
|
||||
git clone -b master git@github.com:open-mmlab/mmdeploy.git MMDeploy
|
||||
cd MMDeploy
|
||||
export MMDEPLOY_DIR=$(pwd)
|
||||
git submodule update --init --recursive
|
||||
|
@ -154,7 +154,7 @@ pip install -e .
|
|||
当前,SDK支持以下设备,
|
||||
|
||||
| 设备 | 名称 | 查找路径 |
|
||||
|:-----|:-----|:-----------------------------------|
|
||||
| :--- | :--- | :--------------------------------- |
|
||||
| Host | cpu | N/A |
|
||||
| CUDA | cuda | CUDA_TOOLKIT_ROOT_DIR 和 pplcv_DIR |
|
||||
|
||||
|
@ -169,7 +169,7 @@ pip install -e .
|
|||
下表展示了目前SDK支持的后端,以及构建时,每个后端需要设置的变量。
|
||||
|
||||
| 推理引擎 | 名称 | 查找路径 |
|
||||
|:------------|:---------|:-------------------------|
|
||||
| :---------- | :------- | :----------------------- |
|
||||
| PPL.nn | pplnn | pplnn_DIR |
|
||||
| ncnn | ncnn | ncnn_DIR |
|
||||
| ONNXRuntime | ort | ONNXRUNTIME_DIR |
|
||||
|
|
|
@ -102,12 +102,12 @@ html_theme_path = [pytorch_sphinx_theme.get_html_theme_path()]
|
|||
# documentation.
|
||||
#
|
||||
html_theme_options = {
|
||||
# TODO: Add log_url
|
||||
# 'logo_url': 'https://mmocr.readthedocs.io/en/latest/',
|
||||
'logo_url':
|
||||
'https://mmdeploy.readthedocs.io/en/latest/',
|
||||
'menu': [
|
||||
{
|
||||
'name': 'GitHub',
|
||||
'url': 'https://github.com/grimoire/deploy_prototype'
|
||||
'url': 'https://github.com/open-mmlab/mmdeploy'
|
||||
},
|
||||
{
|
||||
'name':
|
||||
|
|
Loading…
Reference in New Issue