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