YOLOv4网络加入LSTM后训练报错

错误如下

环境：CUDA 11.2 CUDNN 8.1 Pytorch 1.8.0 利用AlexeyAB的YOLOv4网络进行的训练，在模型中加入了LSTM结构，便出现这种错误，不加LSTM便不报错