我一直在尝试按照指南使用PyTorch为Yolact进行训练：https://github.com/dbolya/yolact
当前的GPU是RTX2070，使用11.1.1的cudatoolkit。
当我运行以下程序时：
python train.py--config=yolact_base_config--batch_size=8
我一直遇到这个错误
RuntimeError：CUDA内存不足。尝试分配50.00 MiB(GPU 0；7.79GiB总容量；5.98 GiB已分配；22.88无MiB；PyTorch总共保留6.40 GiB(
我已经尝试在config.py下减小max_size中图片的最大大小，但CUDA错误仍然存在。我也减少了批量大小，但似乎没有改善。
出现此错误有几个原因，其中一些是
- 你的模型层太密集，所以你的GPU无法训练它
- 图像输入可能太大，GPU无法处理
可能是其他一些原因，你可以尝试的是，查看你的配置，使批量较小。如果你使用8，然后尝试将其减少到4或3或其他什么，看看你的训练在上的效果
或
如果图像。。。或者我试两种