1.expected type torch.FloatTensor but got torch.cuda.FloatTensor
查看了代码,是有个初始化的tensor是在cpu上的,相加的另一个tensor是在cuda上的。
修改:增加.cuda()
initial_hidden_state = initial_hidden_state.cuda()
1.expected type torch.FloatTensor but got torch.cuda.FloatTensor
查看了代码,是有个初始化的tensor是在cpu上的,相加的另一个tensor是在cuda上的。
修改:增加.cuda()
initial_hidden_state = initial_hidden_state.cuda()