open-mmlab / mmaction2

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark
https://mmaction2.readthedocs.io
Apache License 2.0
4.06k stars 1.2k forks source link

get_flops.py[Bug] #2772

Open ChenZhenGui opened 6 months ago

ChenZhenGui commented 6 months ago

Branch

main branch (1.x version, such as v1.0.0, or dev-1.x branch)

Prerequisite

Environment

D:\pythonProject\mmdeploy\venve\Scripts\python.exe D:/ChromeDownload/mmaction2/mmaction/utils/collect_env.py sys.platform: win32 Python: 3.9.5 (tags/v3.9.5:0a7dcbd, May 3 2021, 17:27:52) [MSC v.1928 64 bit (AMD64)] CUDA available: True numpy_random_seed: 2147483648 GPU 0: NVIDIA GeForce GTX 1650 CUDA_HOME: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4 NVCC: Cuda compilation tools, release 11.4, V11.4.100 MSVC: 用于 x64 的 Microsoft (R) C/C++ 优化编译器 19.29.30143 版 GCC: n/a PyTorch: 1.8.0+cu111 PyTorch compiling details: PyTorch built with:

TorchVision: 0.9.0+cu111 OpenCV: 4.7.0 MMEngine: 0.7.3 MMAction2: 1.2.0+4d6c934 MMCV: 2.0.0 MMDetection: 3.0.0 MMPose: 1.0.0

Describe the bug

I tried to use get_flops.py to count the number of parameters and computations in the model, but I made a mistake

Reproduces the problem - code sample

No response

Reproduces the problem - command or script

python tools/analysis_tools/get_flops.py --config 'configs/skeleton/stgcnpp/stgcnpp_8xb16-joint-motion-u100-80e_ntu60-xsub-keypoint-2d.py'

Reproduces the problem - error message

image image

Additional information

No response