Pytork LSTM ValueError:NesteDiOfConction不知道如何处理torch.cuda.FloatTens类型的输入对象

2024-05-23 16:34:16 发布

您现在位置:Python中文网/ 问答频道 /正文

我通过在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的领域


Tags: 方法代码文本机器gpu错误pytorchcpu