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