深度学习 反向传播的时候出现 bad allocation异常

最近在复现TrajGRU,结果跑到一半突然就报错了，不知道怎么解决
希望各位指教该怎么解决这个问题