RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM
这个错误是由于数据类型不一致造成的:
x_emb_embedding(x) #[83, 8, 300] max_len, batch, dim
x_emb=self.dropout(x_emb)
x_emb = (torch.float32)
total_length = x_emb.size(0) #句子的最大长度
packed_embpack_padded_sequence(x_emb, x_len) #[110, 300]即:
encoder_outputs, encoder_hidden = (packed_emb)
这里最后一句出错
因为需求是torch.float32,但是输入是torch.float64,所以会出现错误。
将x_emb转换为torch.float32类型就可以了:x_emb = (torch.float32)
本文发布于:2024-02-05 09:09:44,感谢您对本站的认可!
本文链接:https://www.4u4v.net/it/170728516065193.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
留言与评论(共有 0 条评论) |