RuntimeError: cudnn RNN backward can only be called in training mode

遇到上述狀況model 檢查model.train()以後,是否進行了model.eval(),eval 若是是的話,在下次訓練以前再進行model.eval()便可
相關文章
相關標籤/搜索