mmdeploy/docs/tutorials/how_to_test_model.md

44 lines
1.6 KiB
Markdown
Raw Normal View History

## How to test model
After we convert a PyTorch model to a backend model, we may need to test the speed of the model before using it. In MMDeploy, we provide a tool to test the speed of backend models in `tools/test.py`
## Prerequisite
Before test the speed of a model of a specific backend, you should [install the plugins](../build.md) of the backend and convert the model to the backend with our [deploy tools](how_to_convert_model.md).
## Usage
```shell
python tools/test.py \
${DEPLOY_CFG} \
${MODEL_CFG} \
--model ${BACKEND_MODEL_FILES} \
[--speed-test] \
[--warmup ${WARM_UP}] \
[--log-interval ${LOG_INTERVERL}] \
[--log2file ${LOG_RESULT_TO_FILE}] \
```
## Description of all arguments
* `deploy_cfg`: The config for deployment.
* `model_cfg`: The config of the model in OpenMMLab codebases.
* `--model`: The backend model files. For example, if we convert a model to ncnn, we need to pass a ".param" file and a ".bin" file. If we convert a model to TensorRT, we need to pass the model file with ".engine" suffix.
* `--speed-test`: Whether to activate speed test.
* `--warmup`: warmup before counting inference elapse, require setting speed-test first.
* `--log-interval`: The interval between each log, require setting speed-test first.
* `--log2file`: Log speed test result in file format, need speed-test first.
\* Other arguments in `tools/test.py` are used for performance test. They have no concern with speed test.
## Example
```shell
python tools/test.py \
configs/mmcls/classification_onnxruntime_static.py \
{MMCLS_DIR}/configs/resnet/resnet50_b32x8_imagenet.py \
--model model.onnx \
--speed-test \
--device cpu \
```