if input.size(0) != target.size(0):
IndexError: dimension specified as 0 but tensor has no dimensions
原因:train或者validation dataset 样本数量与batchsize不是整除的关系
解决:调整为整除关系,不再报错(从最大batchsize递减慢慢测试出来的)
if input.size(0) != target.size(0):
IndexError: dimension specified as 0 but tensor has no dimensions
原因:train或者validation dataset 样本数量与batchsize不是整除的关系
解决:调整为整除关系,不再报错(从最大batchsize递减慢慢测试出来的)