Я изучаю RNN и генерацию текста, пытаясь реализовать простой генератор символов с использованием LSTM в PyTorch. Я попытался запустить свой тренировочный код на процессоре, и метод обучения, кажется, работает нормально, но когда я пытаюсь переключиться на другую машину и тренироваться на ее 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?