Expected more than 1 value per channel when training, got input size torch.Size([1, **])-CSDN博客

网站介绍:文章浏览阅读2.4k次,点赞2次,收藏6次。目录出现这种问题大概有两种情况如果上面步骤三还没解决问题原因分析先定位下问题出现这种问题大概有两种情况第一种情况是训练时候出现,另一种是测试模型的时候出现出现这种问题一般是你输入的数据只有一张图片例如 [1,3,256,256]步骤1 你试下改成 [2,3,256,256] 能否解决(如果无法加,请看下一个步骤),如果你是在训练模型,看步骤2 如果你是在测试模型,看步骤3步骤2 如果你是在训练模型,在获取数据集时,将DataLoader中drop_las..._expected more than 1 value per channel when training, got input size torch.s