解决报错:ValueError: Expected input batch_size (10) to match target batch_size (1).

原做的是二分类,在训练的时候遇到了这个报错。

 for i, (inputs, target) in enumerate(trainloader):
            # forward
            output = net(inputs)
            # compute loss
            loss = criterion(output, target)
            optimizer.zero_grad()
            # backward
            loss.backward()
            optimizer.step()

就这样一个简单的训练过程,可是在loss = criterion(output,target)中报错:

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

        print出target:

            tensor([[0],
                        [0],
                        [0],
                        [0],
                        [0],
                        [1],
                        [0],
                        [0],
                        [0],
                        [0]])

在criterion之前将target转为1维: 

target = target.view(1, -1)  # 转换为1维

        结果:tensor([[0, 0, 0, 0, 0, 1, 0, 0, 0, 0]])。

还是报错:ValueError: Expected input batch_size (10) to match target batch_size (1).

这里可以看到结果和一般label的形式不太一样,tensor([[]])多了一个框,一般是tensor([]),这里将target降维处理:

target = target.squeeze()

        结果:tensor([0, 0, 0, 0, 0, 1, 0, 0, 0, 0])。

这样就没有报错了,至于为什么要出现这样,可能是之前在生成label的时候使用了np.array(),将label转成了二维的数组。

  • 7
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值