【算法】CenterNet运行demo时报错：RuntimeError: CUDA error: out of memory

问题描述

python demo.py ctdet --demo /home/vincent/Code/CenterNet-master/images/ --load_model /home/vincent/Code/CenterNet-master/models/ctdet_coco_dla_2x.pth 

运行以上面的demo代码时出现下面的错误（注意文件地址要换成你的地址）

Fix size testing.
training chunk_sizes: [32]
The output will be saved to  /home/vincent/Code/CenterNet-master/src/lib/../../exp/ctdet/default
heads {
   'hm': 80, 'wh': 2, 'reg': 2}
Creating model...
loaded /home