MMAction2常用脚本

计算模型参数量

# --shape:输入张量的形状,格式为 batch_size channels frames height width,如 1 3 32 224 224
python tools/analysis_tools/get_flops.py configs/recognition/swin/swin_tiny_patch244_window877_kinetics400_1x16x1_22k.py --shape 1 3 32 224 224

你可能感兴趣的:(python,深度学习,开发语言)