RuntimeError:库达内存不足。使用Yolact训练时

Question

我一直试图与PyTorch一起为Yolact训练，遵循指南：https://github.com/dbolya/yolact

目前GPU采用RTX2070，cudatoolkit为11.1.1。

当我运行以下命令时：

python train.py --config=yolact_base_config --批处理_train.py=8

我不断地遇到这个错误

RuntimeError:库达没记忆了。尝试分配50.00 MiB (GPU 0；7.79 GiB总容量；5.98 GiB已分配；22.88 MiB空闲；PyTorch总共预留6.40 GiB )

我尝试在max_size中缩小config.py下图片的最大大小，但是数据自动化系统的错误仍然存在。我也减少了批量大小，但似乎没有改善。

Answer 1

造成这一错误的原因有几个，其中一些是

1. 您的模型层太密集，所以您的GPU无法训练它
2. 图像输入可能太大，使得GPU无法处理

。

也许你可以尝试的其他原因是，看看你的心里话，如果你用的是8，那么试着把它减少到4或3，看看你的训练从哪里开始。

或

你可以试着缩小输入尺寸如果图像..。或者我都试一试