还是报错：RuntimeError: CUDA error: device-side assert triggered

这个错误通常是由于CUDA设备上的某些断言失败而引起的。这可能是由于许多原因引起的，例如内存不足、算术溢出或其他未知错误。为了解决这个问题，你可以尝试以下几个步骤：
1. 确保你的CUDA驱动程序和CUDA工具包是最新的版本。
2. 尝试减小batch_size，以减少显存的使用。
3. 尝试减小学习率，以减少训练过程中的波动。
4. 尝试使用其他优化器，例如SGD，而不是Adam。
5. 检查你的代码是否存在任何错误，例如除以零或NaN值。
6. 尝试使用其他主干网络或neck部分，以查看是否存在特定于网络结构的问题。