search

open-mmlab / mmdeploy

OpenMMLab Model Deployment Framework
https://mmdeploy.readthedocs.io/en/latest/
Apache License 2.0
2.75k stars 630 forks source link

When run the profiler.py[Bug] #1588

Closed ChenZhenGui closed 1 year ago

ChenZhenGui commented 1 year ago

Checklist

Describe the bug

When I run the profiler.py by the .pth model,there are some issue

Reproduction

python tools/profiler.py ./configs/mmpose/pose-detection_tensorrt_static-256x192.py /home/nvidia/Desktop/mmpose/configs/body_2d_keypoint/topdown_heatmap/coco/mobilevit_coco-256x192.py /home/nvidia/Desktop/coco/val2017 /home/nvidia/Desktop/mmdeploy/AP_epoch_200.pth

Environment

12/22 18:12:04 - mmengine - INFO - 

12/22 18:12:04 - mmengine - INFO - **********Environmental information**********
/home/nvidia/miniforge3/envs/mobilevit_test/lib/python3.6/site-packages/torchvision/io/image.py:11: UserWarning: Failed to load image Python extension: 
  warn(f"Failed to load image Python extension: {e}")
12/22 18:12:06 - mmengine - INFO - sys.platform: linux
12/22 18:12:06 - mmengine - INFO - Python: 3.6.15 | packaged by conda-forge | (default, Dec  3 2021, 19:12:04) [GCC 9.4.0]
12/22 18:12:06 - mmengine - INFO - CUDA available: True
12/22 18:12:06 - mmengine - INFO - numpy_random_seed: 2147483648
12/22 18:12:06 - mmengine - INFO - GPU 0: NVIDIA Tegra X1
12/22 18:12:06 - mmengine - INFO - CUDA_HOME: /usr/local/cuda-10.2
12/22 18:12:06 - mmengine - INFO - NVCC: Cuda compilation tools, release 10.2, V10.2.300
12/22 18:12:06 - mmengine - INFO - GCC: gcc (Ubuntu/Linaro 7.5.0-3ubuntu1~18.04) 7.5.0
12/22 18:12:06 - mmengine - INFO - PyTorch: 1.8.0
12/22 18:12:06 - mmengine - INFO - PyTorch compiling details: PyTorch built with:
  - GCC 7.5
  - C++ Version: 201402
  - OpenMP 201511 (a.k.a. OpenMP 4.5)
  - NNPACK is enabled
  - CPU capability usage: NO AVX
  - CUDA Runtime 10.2
  - NVCC architecture flags: -gencode;arch=compute_53,code=sm_53;-gencode;arch=compute_62,code=sm_62;-gencode;arch=compute_72,code=sm_72
  - CuDNN 8.2.1
    - Built with CuDNN 8.0
  - Build settings: BLAS_INFO=open, BUILD_TYPE=Release, CUDA_VERSION=10.2, CUDNN_VERSION=8.0.0, CXX_COMPILER=/usr/bin/c++, CXX_FLAGS= -Wno-deprecated -fvisibility-inlines-hidden -DUSE_PTHREADPOOL -fopenmp -DNDEBUG -DUSE_XNNPACK -O2 -fPIC -Wno-narrowing -Wall -Wextra -Werror=return-type -Wno-missing-field-initializers -Wno-type-limits -Wno-array-bounds -Wno-unknown-pragmas -Wno-sign-compare -Wno-unused-parameter -Wno-unused-variable -Wno-unused-function -Wno-unused-result -Wno-unused-local-typedefs -Wno-strict-overflow -Wno-strict-aliasing -Wno-error=deprecated-declarations -Wno-stringop-overflow -Wno-psabi -Wno-error=pedantic -Wno-error=redundant-decls -Wno-error=old-style-cast -fdiagnostics-color=always -faligned-new -Wno-unused-but-set-variable -Wno-maybe-uninitialized -fno-math-errno -fno-trapping-math -Werror=format -DMISSING_ARM_VST1 -DMISSING_ARM_VLD1 -Wno-stringop-overflow, FORCE_FALLBACK_CUDA_MPI=1, LAPACK_INFO=open, TORCH_VERSION=1.8.0, USE_CUDA=ON, USE_CUDNN=ON, USE_EIGEN_FOR_BLAS=ON, USE_EXCEPTION_PTR=1, USE_GFLAGS=OFF, USE_GLOG=OFF, USE_MKL=OFF, USE_MKLDNN=OFF, USE_MPI=ON, USE_NCCL=0, USE_NNPACK=ON, USE_OPENMP=ON, 

12/22 18:12:06 - mmengine - INFO - TorchVision: 0.11.3
12/22 18:12:06 - mmengine - INFO - OpenCV: 4.6.0
12/22 18:12:06 - mmengine - INFO - MMEngine: 0.3.1
12/22 18:12:06 - mmengine - INFO - MMCV: 2.0.0rc2
12/22 18:12:06 - mmengine - INFO - MMCV Compiler: GCC 7.5
12/22 18:12:06 - mmengine - INFO - MMCV CUDA Compiler: 10.2
12/22 18:12:06 - mmengine - INFO - MMDeploy: 0.10.0+71344b2
12/22 18:12:06 - mmengine - INFO - 

12/22 18:12:06 - mmengine - INFO - **********Backend information**********
12/22 18:12:09 - mmengine - INFO - onnxruntime: None    ops_is_avaliable : False
12/22 18:12:10 - mmengine - INFO - tensorrt: 8.0.1.6    ops_is_avaliable : True
12/22 18:12:10 - mmengine - INFO - ncnn: None   ops_is_avaliable : False
12/22 18:12:10 - mmengine - INFO - pplnn_is_avaliable: False
12/22 18:12:10 - mmengine - INFO - openvino_is_avaliable: False
12/22 18:12:10 - mmengine - INFO - snpe_is_available: False
12/22 18:12:10 - mmengine - INFO - ascend_is_available: False
12/22 18:12:10 - mmengine - INFO - coreml_is_available: False
12/22 18:12:10 - mmengine - INFO - 

12/22 18:12:10 - mmengine - INFO - **********Codebase information**********
12/22 18:12:10 - mmengine - INFO - mmdet:   3.0.0rc3
12/22 18:12:10 - mmengine - INFO - mmseg:   None
12/22 18:12:10 - mmengine - INFO - mmcls:   None
12/22 18:12:10 - mmengine - INFO - mmocr:   None
12/22 18:12:10 - mmengine - INFO - mmedit:  None
12/22 18:12:10 - mmengine - INFO - mmdet3d: None
12/22 18:12:10 - mmengine - INFO - mmpose:  1.0.0rc0
12/22 18:12:10 - mmengine - INFO - mmrotate:    None
12/22 18:12:10 - mmengine - INFO - mmaction:    None

Error traceback

Traceback (most recent call last):
  File "tools/profiler.py", line 168, in <module>
    main()
  File "tools/profiler.py", line 153, in main
    model.test_step(data)
  File "/home/nvidia/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 948, in __getattr__
    type(self).__name__, name))
AttributeError: 'TorchWrapper' object has no attribute 'test_step'
AllentDan commented 1 year ago

Close since it is a duplicate of #1569.