0
点赞
收藏
分享

微信扫一扫

RuntimeError: Expected hidden[0] size (1, 64, 256), got (64, 256)

IT程序员 2022-04-06 阅读 94
LSTMexpected

构建双向LSTM网络之后,可能会出现这个问题,原来的写法是这样:

init_hidden = torch.zeros(2, len(x), hidden_size).cuda()

修改为这个样子:

init_hidden = (torch.zeros(2, len(x), hidden_size).cuda(),torch.zeros(2, len(x), hidden_size).cuda())
举报

相关推荐

JWT RS256 base64 md5 解释

0 条评论