pytorch1.6 使用在含有RNN/LSTM网络中使用amp,出错:CUDNN_STATUS_BAD_PARAM,
withtorch.cuda.amp.autocast():preds=model(inp)batch_size=inp.size(0)text,length=converter.encode(labels)preds_size=torch.IntTensor([preds.size(0)]*batch_size)loss=criterion(preds,text,preds_size,lengt