RuntimeError: cuDNN error: CUDNN

阅读: 评论:0

RuntimeError: cuDNN error: CUDNN

RuntimeError: cuDNN error: CUDNN

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_emb&#pack_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小时内删除。

标签:cuDNN   RuntimeError   CUDNN   error
留言与评论(共有 0 条评论)
   
验证码:

Copyright ©2019-2022 Comsenz Inc.Powered by ©

网站地图1 网站地图2 网站地图3 网站地图4 网站地图5 网站地图6 网站地图7 网站地图8 网站地图9 网站地图10 网站地图11 网站地图12 网站地图13 网站地图14 网站地图15 网站地图16 网站地图17 网站地图18 网站地图19 网站地图20 网站地图21 网站地图22/a> 网站地图23