When reasoning about codetr, the program was killed

[Clichong]

Thanks for your error report and we appreciate it a lot.

Checklist

1. I have searched related issues but cannot get the expected help.
2. I have read the FAQ documentation but cannot get the expected help.
3. The bug has not been fixed in the latest version.

Describe the bug
A clear and concise description of what the bug is.

Reproduction

1. What command or script did you run?

python demo/image_demo.py \
   /xxx/xxx/xxx/dataset/images   \
   xxx/best.pth \
   --out-dir xxx/codetr_pred \
  --batch-szie 4  \
  --device "cuda:2"

2. Did you make any modifications on the code or config? Did you understand what you have modified?

I only add some time statistics.Here is the modified I do. Before the data set is fully prepared and reasoning is completed, it is displayed: 1576 killed.

4. What dataset did you use?

Environment

There is no problem with the environment, because my program can be executed normally

Error traceback

Before the data set is fully prepared and reasoning is completed, it is displayed: 1576 killed.