[Bug] the tool(pytorch2onnx) will break when convert basicvsr_reds4 model

[xihao-1223]

Prerequisite

• [X] I have searched Issues and Discussions but cannot get the expected help.
• [X] I have read the FAQ documentation but cannot get the expected help.
• [X] The bug has not been fixed in the latest version (master) or latest version (1.x).

Task

I'm using the official example scripts/configs for the officially supported tasks/models/datasets.

Branch

master branch https://github.com/open-mmlab/mmediting

Environment

Traceback (most recent call last): File "/workspace/xhao/test_env/pymodels/mmediting/mmediting/tools/pytorch2onnx.py", line 232, in data = test_pipeline(data) File "/workspace/xhao/test_env/pymodels/mmediting/mmediting/mmedit/datasets/pipelines/compose.py", line 42, in call data = t(data) File "/workspace/xhao/test_env/pymodels/mmediting/mmediting/mmedit/datasets/pipelines/augmentation.py", line 1093, in call clip_name = results['key'] KeyError: 'key'

Reproduces the problem - code sample

Traceback (most recent call last): File "/workspace/xhao/test_env/pymodels/mmediting/mmediting/tools/pytorch2onnx.py", line 232, in data = test_pipeline(data) File "/workspace/xhao/test_env/pymodels/mmediting/mmediting/mmedit/datasets/pipelines/compose.py", line 42, in call data = t(data) File "/workspace/xhao/test_env/pymodels/mmediting/mmediting/mmedit/datasets/pipelines/augmentation.py", line 1093, in call clip_name = results['key'] KeyError: 'key'

Reproduces the problem - command or script

parser.add_argument('--config', type=str, default="/workspace//test_env/pymodels/mmediting/mmediting/configs/restorers/basicvsr/basicvsr_reds4.py", help='test config file path') parser.add_argument('--checkpoint', type=str, default="/workspace//test_env/pymodels/mmediting/mmediting/weights/basicvsr_reds4_20120409-0e599677.pth", help='checkpoint file')

Reproduces the problem - error message

Traceback (most recent call last): File "/workspace/xhao/test_env/pymodels/mmediting/mmediting/tools/pytorch2onnx.py", line 232, in data = test_pipeline(data) File "/workspace/xhao/test_env/pymodels/mmediting/mmediting/mmedit/datasets/pipelines/compose.py", line 42, in call data = t(data) File "/workspace/xhao/test_env/pymodels/mmediting/mmediting/mmedit/datasets/pipelines/augmentation.py", line 1093, in call clip_name = results['key'] KeyError: 'key'

Additional information

if i want convert a basicvsr_reds4 model to onnx, i need to do ? i do some follow the get_start.md but get an error