cuDNN error: CUDNN"/>
RuntimeError: cuDNN error: CUDNN
RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM
这个错误是由于数据类型不一致造成的:
x_emb=self.gen_embedding(x) #[83, 8, 300] max_len, batch, dim
x_emb=self.dropout(x_emb)
x_emb = x_emb.to(torch.float32)
total_length = x_emb.size(0) #句子的最大长度
packed_emb=torch.nn.utils.rnn.pack_padded_sequence(x_emb, x_len) #[110, 300]即:
encoder_outputs, encoder_hidden = self.rnn(packed_emb)
这里最后一句出错
因为需求是torch.float32,但是输入是torch.float64,所以会出现错误。
将x_emb转换为torch.float32类型就可以了:x_emb = x_emb.to(torch.float32)
更多推荐
RuntimeError: cuDNN error: CUDNN
发布评论