搜尋結果
2024年5月7日 · 作者： [db:作者] 时间：2021-08-18 09:47. 问题描述： RuntimeError: cudnn RNN backward can only be called in training mode. 问题分析： 原因是反向传播的时候不能是非train状态。 我的代码里面在反向传播之前加了个Alexnet，Alexnet计算loss后会自己反向传播，此时在整体LSTM反向传播的时候就会出现这个非train状态的问题。 解决： 百度一下： 第一页的方法我都试过，全都没有用。 彻底解决： 在开始训练之前加上： ? 1. <code>torch.backends.cudnn.enabled=False</code> 即可彻底解决！ 更新：
