使用 24G A5000微调显示显存不足

[ShuweiShao]

感谢作者的复现，我尝试在KITTI数据集上微调，直接运行的作者复现的training代码，不同的地方是我修改了下dataset，但是总是显示 torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 23.67 GiB total capacity; 23.15 GiB already allocated; 10.25 MiB free; 23.27 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF，而且无论怎么调低分辨率都会显示out of memory，请问作者知道是什么原因吗？

另外作者方便加一下微信不，我的微信号：shaoshuweifighting， 十分感谢。