RuntimeError: cudnn RNN backward can only be called in training mode
RuntimeError: cudnn RNN backward can only be called in training mode1 没有用 ‘.train()’ ,有些操作在 ‘.eval()’ 时是被禁止的2 在使用lstm时报的错,加一行下面的代码就可以了torch.backends.cudnn.enabled = False
原创
2021-08-27 07:46:22 ·
668 阅读 ·
3 评论