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

阅读 94

2022-04-06

构建双向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())

精彩评论(0)

0 0 举报