算法CenterNet运行demo异常：CUDA内存不足
发布时间
阅读量：
阅读量
问题描述
python demo.py ctdet --demo /home/vincent/Code/CenterNet-master/images/ --load_model /home/vincent/Code/CenterNet-master/models/ctdet_coco_dla_2x.pth
在执行上述示例代码的过程中，出现了以下所示的错误提示（请注意将文件路径修改为实际使用的路径）
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/vincent/Code/CenterNet-master/models/ctdet_coco_dla_2x.pth, epoch 230
THCudaCheck FAIL file=/opt/conda/conda-bld/全部评论 (0)
还没有任何评论哟~