hjtsml_future-CSDN博客

原创解决报错：ValueError: Expected input batch_size (10) to match target batch_size (1).

原做的是二分类，在训练的时候遇到了这个报错。 for i, (inputs, target) in enumerate(trainloader): # forward output = net(inputs) # compute loss loss = criterion(output, target) optimizer.zero_grad() # back

2021-12-08 09:49:15 11453

原创解决Dataloader报错:ValueError: too many values to unpack (expected 2)

原始代码：# 构建一个能够加载数据集的类class MyDataset(Dataset): # 继承Dataset # 得到一个包含图像和标签的列表 def __init__(self, root, label_root,transform=None,is_aug=None,fix_size=(244, 244)): # 初始化一些属性 data_list = get_data(root) # 得到数据的名字列表 label = get_labe

2021-12-08 09:25:09 3808 1

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

原创 解决报错：ValueError: Expected input batch_size (10) to match target batch_size (1).

原创 解决Dataloader报错:ValueError: too many values to unpack (expected 2)

空空如也

空空如也

原创解决报错：ValueError: Expected input batch_size (10) to match target batch_size (1).

原创解决Dataloader报错:ValueError: too many values to unpack (expected 2)