RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM

╰+哭是因爲堅強的太久メ 2023-02-23 07:47 82阅读 0赞 
  1. RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM

这个错误是由于数据类型不一致造成的：

  1. x_emb=self.gen_embedding(x) #[83, 8, 300] max_len, batch, dim
  2. x_emb=self.dropout(x_emb)
  3. x_emb = x_emb.to(torch.float32)
  4. total_length = x_emb.size(0) #句子的最大长度
  5. packed_emb=torch.nn.utils.rnn.pack_padded_sequence(x_emb, x_len) #[110, 300]即：
  6. encoder_outputs, encoder_hidden = self.rnn(packed_emb)

这里最后一句出错
因为需求是torch.float32,但是输入是torch.float64，所以会出现错误。
将x_emb转换为torch.float32类型就可以了：x_emb = x_emb.to(torch.float32)

发表评论

表情：
评论列表 （有 0 条评论，82人围观）

还没有评论，来说两句吧...

相关阅读