PytorchのRuntime errorについて

Question

titanicのデータでPytorchの練習をしていたところ学習でエラーが出てしまいました。
訓練データ(train_x):(891,13)
訓練ラベル(train_y):(891,1)
テストデータ(test_x):(418,13)　の大きさで下記のコードを書きました。
試したこととして訓練ラベルをTensorに変換してみたりもしましたがエラーは変わりませんでした。改善策はありますでしょうか？

train_x = torch.Tensor(data_x.values)
train_y = torch.LongTensor(data_y.values)
test_x = torch.Tensor(test.values)

optimizer = optim.SGD(model.parameters(), lr=0.02)
train_loss = []
train_accu = []
i = 0

model.train() #学習モードに切り替え

for epoch in range(2000): # 数字は適当
    data, target = Variable(train_x), Variable(train_y)
    optimizer.zero_grad()
    output = model(data) 
        
    loss = F.nll_loss(output, target)
    loss.backward()
    train_loss.append(loss.data.item())
    optimizer.step()
        
    prediction = output.data.max(1)[1]
    accuracy = prediction.eq(target.data).sum().numpy() / len(train_x)
    train_accu.append(accuracy)
    
    if i % 10 == 0:
        print('Train Step: {}\tLoss: {:.3f}\tAccuracy: {:.3f}'.format(i, loss.data.item(), accuracy))
    i += 1

print('Train Step: {}\tLoss: {:.3f}\tAccuracy: {:.3f}'.format(i, loss.data.item(), accuracy))

model.eval() #推論モードに切り替え
outputs = model(Variable(test_x))
_, predicted = torch.max(outputs.data, 1)

そして以下のエラーが出ました

RuntimeError                              Traceback (most recent call last)
<ipython-input-161-57d29f49e986> in <module>
     11     output = model(data)
     12 
---> 13     loss = F.nll_loss(output, target)
     14     loss.backward()
     15     train_loss.append(loss.data.item())

~/opt/anaconda3/lib/python3.7/site-packages/torch/nn/functional.py in nll_loss(input, target, weight, size_average, ignore_index, reduce, reduction)
   2216                          .format(input.size(0), target.size(0)))
   2217     if dim == 2:
-> 2218         ret = torch._C._nn.nll_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
   2219     elif dim == 4:
   2220         ret = torch._C._nn.nll_loss2d(input, target, weight, _Reduction.get_enum(reduction), ignore_index)

RuntimeError: 1D target tensor expected, multi-target not supported

Answer 1

時間がたっているので解決されているかもしれませんが・・・。
軽くソースコードを検索したところ、以下のサイトがヒットしたのですが、
https://megane-man666.hatenablog.com/entry/pytorch
こちらのコードの通りにこちらの環境で確認したところ、特に問題なく最後まで走りました。
記載頂いているコードと異なるところもないようですし、記載頂いている場所以外に起因するエラーだと思います。
エラーコードを見る限り、train_xかtrain_yのshapeが間違ってそうなので、記載頂いたコードの前の部分を再度確認してみてください。
pytorchに限りませんが、こういったshapeの違いによるエラーは頻繁に出くわします。疑わしい部分のshapeをprintしたりなど、細かくトラブルシューティングする力は必要になってくると思いますので、頑張って下さい。