RuntimeError: input.size (-1) должен быть равен input_size - PullRequest
Новая
       32

RuntimeError: input.size (-1) должен быть равен input_size

0 голосов
/ 17 июня 2019 
from torch.utils.data import Dataset, DataLoader

if __name__ == '__main__':
x_y_data_set = LyricsDataSet(lyrics)
data_loader = DataLoader(x_y_data_set, batch_size=1,
                         shuffle=True)
trainer.run(data_loader, 10)

RuntimeError                              Traceback (most recent call 
last)
<ipython-input-59-38376625da41> in <module>()
    296     data_loader = DataLoader(x_y_data_set, batch_size=1,
    297                              shuffle=True)
--> 298     trainer.run(data_loader, 10)

14 frames
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/rnn.py in 
check_input(self, input, batch_sizes)
    151             raise RuntimeError(
    152                 'input.size(-1) must be equal to input_size. 
Expected {}, got {}'.format(
--> 153                     self.input_size, input.size(-1)))
    154 
    155     @weak_script_method

RuntimeError: input.size(-1) must be equal to input_size. Expected 300, 
got 96

Код отлично работает в Anaconda, но когда я запускаю его в Google Colab, он выдает RuntimeError. Я не могу найти проблему. Любая помощь будет оценена.

...