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+