lifsgd

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 12, 1])

7

您好！ 我是一名AI学习新手，在一张2070Super复现您代码的过程中，出现了这个问题。 第一次是在第一个epoch快结束时出现，网上查了些，试了batch_size调大，但调大有另外的问题，试了设置drop_last=True，也有其他的问题。 在刚刚按最初的配置运行，完成一次epoch，以为这个问题没有了，结果第二次epoch开始没多久，又出现了。