输入批次只有一个数据点的原因,至少需要如下形式的数据:
input = Variable(torch.FloatTensor(2, 3, 256, 128)),也可以将DataLoader中drop_last设置为True,把不够一个批次的数据丢弃。
ValueError: Expected more than 1 value per channel when training
最新推荐文章于 2024-04-26 15:47:14 发布
输入批次只有一个数据点的原因,至少需要如下形式的数据:
input = Variable(torch.FloatTensor(2, 3, 256, 128)),也可以将DataLoader中drop_last设置为True,把不够一个批次的数据丢弃。