terminate called after throwing an instance of 'std::bad_alloc'

[sihuo521]

error:
terminate called after throwing an instance of 'std::bad_alloc'
what(): std::bad_alloc
infer.sh: line 1: 21934 Aborted (core dumped)

shell:
./libtorch-yolov5 --source path/*.jpg --weights path/best.torchscript --gpu --view-img