BUG:在测试的时候,传递prev_state_spatial,出现显存溢出问题。
解决方案,将模型得到的中间变量保存至cpu,在使用的时候,将该数据转移到cuda上。
参考文献:
【1】https://blog.csdn.net/nlite827109223/article/details/89847044
BUG:在测试的时候,传递prev_state_spatial,出现显存溢出问题。
解决方案,将模型得到的中间变量保存至cpu,在使用的时候,将该数据转移到cuda上。
【1】https://blog.csdn.net/nlite827109223/article/details/89847044