我通过在PyTorch中使用LSTM实现一个简单的字符生成器来学习RNN和文本生成。我曾尝试在CPU上运行我的培训代码,培训方法似乎运行良好,但当我尝试切换到另一台机器并在其GPU上培训时,我遇到以下错误:
ValueError: NestedIOFunction doesn't know how to process an input object of type torch.cuda.FloatTensor
这在使用output, h = net(Variable(inputs), h)
获取输出和隐藏层时发生。我已经将inputs
转换为它的.cuda()
对应物,我似乎找不到可能引发此错误的原因。是否还有另一个我可能忘记转换为cuda
的领域
目前没有回答
相关问题 更多 >
编程相关推荐