ValueError: Expected input batch_size (1) to match target batch_size (0).

Question

ValueError: Expected input batch_size (1) to match target batch_size (0).这个问题该怎么解决呢，找遍全网都没找到合适的答案，头大头大，请神通广大的网友帮忙解决一下，感谢感谢感谢

Answer 1

可以参考

解决ValueError: Expected input batch_size () to match target batch_size ().问题_小破船ZAW的博客-CSDN博客 一般这种问题是由于输入图像的大小不匹配导致，class Network(nn.Module): def __init__(self, height_size, width_size, num_classes): super(Network, self).__init__() self.height_size = height_size self.width_size = width_size self.conv2d_1 = nn.Co https://blog.csdn.net/weixin_55168780/article/details/122886261