雅虎香港 搜尋

搜尋結果

  1. 2024年5月7日 · 如果此时有张量保存在GPU,往往也会先造成显存爆炸!. 问题描述 RuntimeError: cudnn RNN backward can only be called in training mode 问题分析 原因是反向传播的时候不能是非train状态。. 我的代码里面在反向传播之前加了个AlexnetAlexnet计算loss后会自己反向传播此时在整体LSTM ...

  1. 其他人也搜尋了