How to modify the default input size when running get_flops.py？

[4A2008]

The official document seems to be ineffective：
python tools/analysis_tools/get_flops.py ${CONFIG_FILE} [--shape ${INPUT_SHAPE}]

python tools/analysis_tools/get_flops.py /data/ctc/mmdetection/configs/YGdataset/mask-rcnn_r50-caffe_fpn_ms-poly-1x_YGdataset.py --shape 1024 1024
usage: get_flops.py [-h] [--num-images NUM_IMAGES] [--cfg-options CFG_OPTIONS [CFG_OPTIONS ...]] config
get_flops.py: error: unrecognized arguments: --shape 1024 1024

Use size divisor set input shape from (1024, 1024) to (800, 800)

Compute type: dataloader: load a picture from the dataset
Input shape: (800, 800)
Flops: 0.183T
Params: 43.987M