问答详情
源自:4-4 中文文本分类实战(下)

关于多分类任务的一个报错

您好,我在完成一个多分类任务时遇到了一些困难。我运行指令python run.py --model TextCNN --word True --embedding random后报错Traceback (most recent call last):  

File "run.py", line 54, in <module>    train(config, model, train_iter, dev_iter, test_iter)  File "C:\Users\Administrator\Desktop\project\System\TextCNN\train_eval.py", line 47, in train   

 loss = F.cross_entropy(outputs, labels)  

File "C:\Users\Administrator\Desktop\project\System\venv\lib\site-packages\torch\nn\functional.py", line 2996, in cross_entropy    

return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing) 

IndexError: Target 48 is out of bounds。

我在网上看到的解释是输出层数跟我的标签数不匹配,我想问一下我应该改动哪里的代码呢?

提问者:weixin_慕莱坞6512356 2022-05-03 16:30

个回答

  • 慕梦前来
    2022-06-06 21:27:33

    IndexError: Target 48 is out of bounds。这一句已经告诉你在哪儿了,