使用Pytorch训练，报错Runtime Error: cudnn RNN backward can only be called in training mode

直接原因：第一个Epoch训练正常，第一次切换至Eval并进行测试正常，第二个Epochs训练时报错

根本原因：默认Model的模式是

?
1
net.train()

但是测试时需要调成

?
1
net.eval()

因为再次训练时没调回 train 模式，故出现上述 bug， 只需在训练之前加上

?
1
net.train()

即可解决

posted @   TheBigSeven  阅读(1521)  评论(0)  编辑  收藏  举报
相关博文：
· 深度学习报错：Expected more than 1 value per channel when training, got input size torch.Size([1, 256])
· 使用Pytorch运行过去别人深度学习的代码出现：No module named 'tokenizer'
· 踩坑日志1：cuDNN问题UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR
· pytorch报错：Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass RuntimeError: FIND was unable to find an engine to execute this computatio
· RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
阅读排行：
· DeepSeek 开源周回顾「GitHub 热点速览」
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布：重大改进与新特性概览！
· AI与.NET技术实操系列（二）：开始使用ML.NET
· .NET10 - 预览版1新功能体验（一）
点击右上角即可分享
微信分享提示
AI FOR CODE 大赛