运行pytorch时,训练很正常,但是如果切换到eval()模式之后再继续训练, 发现报错:
RuntimeError: cudnn RNN backward can only be called in training mode
原因分析:由于在训练时,设置的是net.train() 切换到预测模式时, 设置为net.eval(), 再回到训练环节,此时的网络依然是eval()模式,因此出现上述bug, 只要在继续训练模型之前加上net.train()即可完美解决问题
运行pytorch时,训练很正常,但是如果切换到eval()模式之后再继续训练, 发现报错:
RuntimeError: cudnn RNN backward can only be called in training mode
原因分析:由于在训练时,设置的是net.train() 切换到预测模式时, 设置为net.eval(), 再回到训练环节,此时的网络依然是eval()模式,因此出现上述bug, 只要在继续训练模型之前加上net.train()即可完美解决问题