202 RuntimeError: cudnn RNN backward can only be called in training mode
时间: 2024-04-03 13:31:53 浏览: 170
Python RuntimeError: thread.__init__() not called解决方法
这个错误通常是由于在 PyTorch 中使用了 cudnn RNN 模块并且模型处于评估模式(eval mode)时发生的。在评估模式下,cudnn RNN 模块无法进行反向传播。解决方法是在训练模式(train mode)下进行模型训练,或者使用 PyTorch 自带的 RNN 模块代替 cudnn RNN 模块。如果你一定要在评估模式下使用 cudnn RNN 模块,可以尝试在模型中添加 `model.train()` 来将模型转换为训练模式,或者在调用 cudnn RNN 模块时将 `training=True`。
阅读全文