下大禹了 2019-01-24 23:08 采纳率: 0%
浏览 7980

pytorch 的Cross Entropy Loss 输入怎么填?

以识别一个四位数的验证码为例,批次取为100,标签用one_hot 表示,则标签的size为[100,4,10],input也为[100,4,10],请问loss用torch.nn.CrossEntropyLoss时,输入的input和target分别应为多少?

另外,用其他几种损失函数时,以四位验证码为例,输入该各是多少?

  • 写回答

2条回答 默认 最新

  • threenewbee 2019-01-25 00:45
    关注

    如果你的验证码是数字,你的损失函数前应该是一个1000个神经元的softmax层,那么输入输出的size都应该是[1000,1]

    评论

报告相同问题?