https://github.com/pytry3g/pytorch-example/tree/master/nlp/rnn/wikipedia
上記githubのtest_model.pyとtrain.pyを実行すると、共に、
以下のエラーが出てしまいます。
Traceback (most recent call last):
File "/home/yudai/デスクトップ/wikipedia/test_model.py", line 53, in <module>
model = RNN(n_vocab, embedding_dim, hidden_dim)
File "/home/yudai/デスクトップ/wikipedia/test_model.py", line 22, in init
self.decoder = nn.Linear(hidden_dim, n_vocab)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/linear.py", line 46, in init
self.reset_parameters()
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/linear.py", line 49, in reset_parameters
stdv = 1. / math.sqrt(self.weight.size(1))
RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1)
本当に困っています。
何卒、よろしくお願い致します。
環境
OS は Ubuntu です。
$ uname -a
Linux NAGARE 4.15.0-30-generic #32-Ubuntu SMP Thu Jul 26 17:42:43 UTC 2018 x86_64 x86_64 x86_64 GNU/Linux
ライブラリ:pytorch
torch 0.4.1
torchvision 0.2.1
$ python3 -V
Python3.6.6+